# 导入训练好的模型
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')