不同版本pytorch训的模型之间的转化

悠悠 2022-10-22 13:55 398阅读 0赞

写在开头: 本文仅记录自己遇到的问题，方便之后查阅。

最近用不同版本的pytorch训了一些模型，但使用pytorch1.3加载pytorch1.6训练的模型失败，显示如下所示等一些乱七八糟的错误。

20210317163611114.png

但是没关系，因为本宝想到了一个好的解决办法，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: 

  1. 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版本之间可以互相加载彼此训练的模型了，简直完美，棒棒哒!

发表评论

表情：
评论列表 （有 0 条评论，398人围观）

还没有评论，来说两句吧...

相关阅读