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

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

 

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