尝试使用 torch.load(PATH) 加载预训练模型时出现 unicode 解码错误答案

【问题标题】：getting unicode decode error while trying to load pre-trained model using torch.load(PATH)尝试使用 torch.load(PATH) 加载预训练模型时出现 unicode 解码错误
【发布时间】：2020-05-17 11:45:40
【问题描述】：

尝试使用 torch.load(PATH) 加载 ResNet 18 预训练模型，但遇到 Unicode 解码错误，请帮忙。

Traceback (most recent call last):
  File "main.py", line 312, in <module>
    main()
  File "main.py", line 138, in main
    checkpoint = torch.load(args.resume)
  File "F:\InsSoft\Anaconda\lib\site-packages\torch\serialization.py", line 593, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
  File "F:\InsSoft\Anaconda\lib\site-packages\torch\serialization.py", line 773, in _legacy_load
    result = unpickler.load()
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xbe in position 2: invalid start byte

【问题讨论】：

您使用的是哪个 python 版本？预训练后更新 pytorch 了吗？
github.com/pytorch/pytorch/issues/5994 可能会有帮助。
@WasiAhmad 是的，pytorch 是在预训练后更新的

标签： unicode anaconda pytorch

【解决方案1】：

每当模型在 Torch 版本 < 0.4 上进行预训练但使用 Torch 版本 > 0.4 进行测试/恢复时，就会出现此错误。
所以使用checkpoint = torch.load(args.resume,encoding='latin1')

【讨论】：