pytorch 加载训练好的模型后测试得到的结果不一样

1、训练模型,在模型train模式下,运行测试图片,得到结果正确:5200240905064,保存此模型,

2、加载训练好的pth模型,调用model.eval(),在eval模式下,去测试我下载的一张图片,发现结果为:5200240905064。

可能原因是train模式和eval模式模型运行结果不同

解决方案:

1、在训练模型进行评估之前,调用model.eval(),让模型工作在eval模式下进行模型测试评估,这样得到的结果才和最终模型推理结果一致。

2、或者直接在训练多一些,本例中我又训练了200epochs后结果一致。

你可能感兴趣的:(pytorch,机器学习,深度学习,人工智能,python)