【发布时间】：2018-02-20 09:29:54
【问题描述】：
我已经在 pytorch 中训练了一个预训练的 RESNET18 模型并保存了它。在测试模型时，针对不同的 mini-batch 大小给出了不同的精度。有谁知道为什么？
【问题讨论】：
【发布时间】：2018-02-20 09:29:54
【问题描述】：
是的，我想是的。 RESNET 包含批量标准化层。在评估时，您需要修复这些；否则，在处理每批后，运行装置会不断调整，从而为您提供不同的精度。
尝试设置：
model.eval()
在评估之前。请注意，在重新开始培训之前，请致电model.train()。
【讨论】：
-
我忘了用答案更新线程。这正是奏效的。谢谢。