标签:info 模型 pytorch dict 版本 model pkl 加载
写在开头: 本文仅记录自己遇到的问题,方便之后查阅。
最近用不同版本的pytorch训了一些模型,但使用pytorch1.3加载pytorch1.6训练的模型失败,显示如下所示等一些乱七八糟的错误。
但是没关系,因为本宝想到了一个好的解决办法,hiahiahia~
1. pytorch1.6 加载它自身生成的模型是没问题的,然后把加载后的模型dict生成pickle文件, 代码如下:
import pickle as pkl
import torch
info_dict = torch.load('model_final.pth')
with open('pkl_model_final.pth', 'wb') as f:
pkl.dump(info_dict, f)
2 pytorch1.3 加载时,使用以下命令:
model = YOUR_MODEL(args)
with open( '/pkl_model_final_pth', 'rb') as f:
info_dict = pkl.load(f)
model.load_state_dict(info_dict['model'])
这样就可以保证不同pytorch版本之间可以互相加载彼此训练的模型了,简直完美,棒棒哒!
标签:info,模型,pytorch,dict,版本,model,pkl,加载 来源: https://blog.csdn.net/u010454261/article/details/114936724
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。