解决'ascii' codec can't decode byte 0xc3 in position 918: ordinal not in range(128)

最新推荐文章于 2023-02-10 16:07:48 发布

larryli007

最新推荐文章于 2023-02-10 16:07:48 发布

阅读量2.3k

点赞数 3

CC 4.0 BY-SA版权

分类专栏：学习笔记

本文链接：https://blog.youkuaiyun.com/larryli007/article/details/85274905

学习笔记专栏收录该内容

6 篇文章

订阅专栏

本文详细介绍了如何解决在Python3环境下加载由Python2环境下训练的PyTorch模型的问题。通过调整pickle模块的加载方式，确保了模型的正确加载。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个问题出现的原因是：所要加载的预训练model是使用python2环境下的pytorch训练的。
解决方法：
通过以下方式加载模型

from functools import partial
import pickle
pickle.load = partial(pickle.load, encoding="latin1")
pickle.Unpickler = partial(pickle.Unpickler, encoding="latin1")
model = torch.load(model_file, map_location=lambda storage, loc: storage, pickle_module=pickle)