转载自:https://www.jianshu.com/p/6ba95579082c
只保存模型参数
torch.save(model.state_dict(), '\parameter.pkl')
model = TheModelClass(...)
model.load_state_dict(torch.load('\parameter.pkl'))
保存完整模型
torch.save(model, '\model.pkl')
model = torch.load('\model.pkl')