python - 在 Pytorch Upgrade 之后，我的模型给出了几乎随机的输出

Question

我在“Pytorch 0.4.1”中训练、测试并仍在使用模型。如果我使用音高 0.4.1，它曾经并且仍然可以正常工作（输出应该是）。但是当我升级到 1.0.1 版时，每次我尝试评估相同的输入图像，我都会得到不同的输出（它是回归）。

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

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

Answer 1

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