美文网首页工作生活
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