标签：训练 pth 模型 torch state pytorch num dict model

1 保存和加载整个模型 

torch.save(model_object, 'model.pth') 
model = torch.load('model.pth')

2 仅保存和加载模型参数

torch.save(model_obj.state_dict(), 'params.pth')  
model_obj.load_state_dict(torch.load('params.pth'))  

3 选择保存网络中的一部分参数或者额外保存其余的参数

torch.save({'state_dict': net.state_dict(), 'linear1':net.linear1.state_dict(),
            'optimizer': optimizer.state_dict(),'num_epoch':num_epochs },
            'detail.pth')

model = torch.load('detail.pth')
net = DNN(num_input,num_hidden1,num_hidden2,num_output)
net.load_state_dict(model['state_dict'])

参考：

[日常] PyTorch 预训练模型，保存，读取和更新模型参数以及多 GPU 训练模型

标签：训练,pth,模型,torch,state,pytorch,num,dict,model
来源： https://www.cnblogs.com/BlairGrowing/p/15981277.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。