【Bug解决】UnpicklingError: A load persistent id instruction was encountered, but no persistent_load.

zstar

发布于 2022-06-14 15:04:16

1.9K0

问题环境

在将pytorch本地训练好的模型，传到jetson nano平台上，加载模型时，报了这个错误：

UnpicklingError: A load persistent id instruction was encountered, but no persistent_load.

查阅相关资料，这是由于pytorch在1.6.0版本之后，模型保存默认方式是压缩的形式，而在jetson nano版本上的pytorch版本是1.0.0，因此无法直接加载.pth文件。用下面的方式可以查看pytorch的版本：

import torch
print(torch.__version__)

设置保存模型时不进行压缩，即修改保存模型语句为：

torch.save(model.state_dict(), MODEL_PATH, _use_new_zipfile_serialization=False)

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2022-06-11，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度