解决pytorch加载模型报错TypeError: ‘collections.OrderedDict‘ object is not callable

最新推荐文章于 2022-08-11 18:09:16 发布

Hello3q3q

最新推荐文章于 2022-08-11 18:09:16 发布

阅读量2w

点赞数 11

CC 4.0 BY-SA版权

分类专栏：异常与错误处理文章标签： pytorch 神经网络 python

26 篇文章

订阅专栏

博客讨论了在使用PyTorch时遇到的一个常见问题，即尝试加载已保存的网络模型时出现错误。错误源于仅保存了网络的参数状态字典，而非整个模型结构。解决方案是首先实例化网络结构，然后通过load_state_dict方法导入参数。示例代码中展示了如何正确加载BiGRU网络的模型和参数。

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

model = torch.load('Net.pkl')

错误信息为：
错误原因：之前保存网络时用的方法是torch.save(model, 'Nei.pkl'),这样保存下来的Net.pkl是一个状态字典，而不是模型本身，也就是说Net.pkl中保存的只是网络的参数，而没有网络结构。
改正方法：应该先载入网络结构，再导入网络的参数，修改后代码如下，其中 BiGRU()是写好的神经网络类。

model = BiGRU()	# 导入网络结构
model.load_state_dict(torch.load('Net.pkl')) # 导入网络的参数

这样就不会报错了！