这是由于模型是在python2训练出来的,由于编码问题在python3导致无法读取。 我们使用pickle做跳转
解决办法:
先在python2:
checkpoint = torch.load("path")
with open("path", 'wb') as f:
pickle.dump(checkpoint,f)
回到py3
pkl_file = open('path','rb')
data = pickle.load(pkl_file, encoding='latin1')
torch.save(data, "path")
网友评论