解决AttributeError: ‘collections.OrderedDict‘ object has no attrib
【摘要】 报错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)