_pickle.UnpicklingError: A load persistent id instruction was encountered...报错信息的解决办法

Introduction 引言

在使用PyTorch训练神经网络模型的时候，由于PyTorch安装版本不同的原因，可能会因为不兼容而报出各种各样的错误，此系列专注于记录在训练时候遇到的各种错误解决方法，供小伙伴们参考和学习。

错误信息描述

话不多说，如题，首先抛出错误信息：

C:\Users\***\python.exe "D:/***/ConfusionMatrix_vgg/main.py"
cuda:0
Traceback (most recent call last):
  File "D:/***/ConfusionMatrix_vgg/main.py", line 107, in <module>
    net.load_state_dict(torch.load(model_weight_path, map_location=device))
  File "C:\Users\***\Anaconda3\envs\pytorch\lib\site-packages\torch\serialization.py", line 367, in load
    return _load(f, map_location, pickle_module)
  File "C:\Users\***\Anaconda3\envs\pytorch\lib\site-packages\torch\serialization.py", line 528, in _load
    magic_number = pickle_module.load(f)
_pickle.UnpicklingError: A load persistent id instruction was encountered,
but no persistent_load function was specified.

Process finished with exit code 1

报错信息是个好东西，你要学会如何从报错信息中发掘出有用的东西来。

报错原因分析

使用PyTorch训练得到网络模型文件后，在验证和混淆矩阵分析阶段，可能会遇到上述报错信息，可能原因是训练阶段和验证阶段的PyTorch版本不一致。

解决方法

将网络模型的训练阶段和测试阶段的PyTorch版本改为一致，即可成功消除该错误。

Conclusion 总结

综上所述，由于网络模型的训练阶段和测试阶段的PyTorch版本不一致，在训练网络模型的时候会出现报错的情况，小伙伴们可以参考上述方法进行改进之。

写到这里，差不多本文就要结束了，如果有问题可以在下方留言区留言交流。如果我的这篇文章帮助到了你，那我也会感到很高兴，一个人能走多远，在于与谁同行。

Reference 参考

[1] 解决：_pickle.UnpicklingError: A load persistent id instruction was encountered,