pytorch==1.6.0(或更高的)的框架下保存模型时,不建议使用
torch.save(model.state_dict(),model_path) # 该语句只保存了模型的权重参数未保存整个模型
model.eval()
加载模型的时候会报错
'collections.OrderedDict' object has no attribute 'eval'
正确语句:
torch.save(model,'save_path') #保存时
torch.load('save_path/model') #加载时
因此只需要修改torch.save语句,再重新训练即可。导致的原因是版本不一样,发生了变化。