python - Pytorch 升级后，我的模型几乎给出了随机输出

Question

我在“Pytorch 0.4.1”中训练、测试并仍在使用模型。如果我使用音高 0.4.1，它过去和现在仍然工作正常(输出应该是这样)。但是当我升级到版本 1.0.1 时，每次我尝试评估相同的输入图像时，都会得到不同的输出(这是一个回归)。

我试图查看这些版本中发生了什么更改，但由于我没有收到任何错误、警告任何内容，所以我不确定应该具体寻找什么。

PS:我检查了权重，加载模型时它们也是相同的

Answer 1

这是因为 drop out 层。 Model.eval 禁用 dropout 层。很简单。但是现在在Pytorh升级中，如果没有在模型init函数中专门定义Dropout，则在eval期间它不会被禁用。至少这是我的案子的原因。