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

引起报错的代码

# 导入训练好的模型
model = torch.load('../179712K82DE1024256RP4700rsnet105Istrip0.9947.pth')

原因

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

 val_accurate = test_accuracy / test_num  # lewen添加2
        if  val_accurate > best_acc:
            best_acc = val_accurate
            torch.save(net.state_dict(), save_path)  # 保存参数
        print("模型已保存")

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

解决方案

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

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