Pytorch 预训练的 RESNET18 模型

Question

我在pytorch中训练了一个预训练好的RESNET18模型并保存了。在测试模型时，针对不同的小批量大小给出了不同的精度。有谁知道为什么？

Answer 1

是的，我想是的。 RESNET 包含批量归一化层。在评估时，您需要修复这些问题；否则 运行 意味着在处理每批次后会不断调整，从而为您提供不同的准确性。

尝试设置：

model.eval()

评估前。请注意，在重新开始训练之前，请致电 model.train().