我在pytorch训练了一个训练有素的RESNET18模型并保存了它 . 虽然测试模型对不同的小批量尺寸给出了不同的精度 . 有谁知道为什么？
是的，我想是这样 . RESNET包含批量标准化层 . 在评估时你需要解决这些问题;否则，在处理每批产品后，运行工具会不断调整，从而为您提供不同的精度 .
尝试设置：
model.eval()
在评估之前 . 请在返回培训之前注意，请致电 model.train() .
model.train()
1 回答
是的，我想是这样 . RESNET包含批量标准化层 . 在评估时你需要解决这些问题;否则，在处理每批产品后，运行工具会不断调整，从而为您提供不同的精度 .
尝试设置：
在评估之前 . 请在返回培训之前注意，请致电
model.train().