解决AttributeError: ‘collections.OrderedDict‘ object has no attrib

举报
AiCharm 发表于 2023/05/14 17:09:15 2023/05/14
【摘要】 报错AttributeError: ‘collections.OrderedDict‘ object has no attribute ‘eval‘ 原因这个错误的原因就是说你这个collecttions的类对象没有eval这个属性。绝大部分原因是因为使用了下面语句来保存模型。torch.save(model.state_dict(),model_path)但实际上它保存的不是模型文件,而...

报错

AttributeError: ‘collections.OrderedDict‘ object has no attribute ‘eval

原因

这个错误的原因就是说你这个collecttions的类对象没有eval这个属性。
绝大部分原因是因为使用了下面语句来保存模型。

torch.save(model.state_dict(),model_path)

但实际上它保存的不是模型文件,而是参数文件文件。在模型文件中,存储完整的模型,而在状态文件中,仅存储参数。因此,collections.OrderedDict只是模型的值。

解决方案

适用下面的语句来保存模型

#保存时
torch.save(model,'save_path')
#加载时
torch.load('save_path/model')

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。