pytorch载入模型时UnicodeDecodeError

这是由于模型是在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")

你可能感兴趣的:(pytorch载入模型时UnicodeDecodeError)