报错
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')
更多Ai资讯:公主号AiCharm