美文网首页工作生活
pytorch载入模型时UnicodeDecodeError

pytorch载入模型时UnicodeDecodeError

作者: 点指兵兵_ | 来源:发表于2019-07-01 15:18 被阅读0次

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

    本文链接:https://www.haomeiwen.com/subject/gxigcctx.html