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

加载训练好的pth模型之后，去测试我下载的一张图片，发现有时候是标签0，有时候是3，有时候是5（正确的是5）。

解决方案：

加入语句：

model.eval()

转换为测试模式，此时不启用 BatchNormalization 和 Dropout，将BatchNormalization和Dropout置为False。否则批量归一化和dropout容易产生较大的波动。