因为训练的pytorch版本是0.4的,测试的pytorch版本是1.2的,所以在测试的时候报错:
解决方法:
新建一个.py文件:
写入如下代码:
import torch
state_dict = torch.load('.pth路径名')
torch.save(state_dict, ".pth路径名", _use_new_zipfile_serialization=False)
将代码在原来的pytorch版本下面运行,将文件转换成非.zip文件,然后就可以在现在的pytorch版本下加载了。