
解决Pytorch加载已训练模型时遇到的错误问题
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文章主要介绍了解决使用Pytorch框架在加载预训练模型过程中常见的各类错误的方法和技巧。
这是一个非常愚蠢的错误,在debug的时候一定要仔细查看error信息,并提醒自己要认真对待这些提示!切记!切记!
在使用PyTorch保存模型和加载模型时,代码其实很简单:
# 保存整个网络和参数
torch.save(your_net, save_name.pkl)
# 加载保存的模型
net = torch.load(save_name.pkl)
因为比较懒,就想直接把整个网络都保存下来,在测试文件中再直接加载。结果却遇到了问题:由于粗心大意没有仔细阅读error信息,将‘Net’误读为‘net’。因此出现了“报错没有属性‘net’”的错误提示。
希望这个经历能够提醒大家在处理这类情况时要更加细心和耐心!
全部评论 (0)
还没有任何评论哟~


