2024-05-01 21:59:38 发布
网友
我在Pythorch中训练了一个预先训练过的RESNET18模型并保存了它。在测试过程中,对于不同的小批量,模型给出了不同的精度。有人知道为什么吗?在
是的,我想是的。 RESNET包含批量标准化层。在评估时,您需要修复这些问题;否则,在处理每个批次之后,运行方式会不断调整,从而给您不同的精度。在
尝试设置:
model.eval()
评估前。注意:在重新开始训练之前,请致电model.train()。在
model.train()
是的,我想是的。 RESNET包含批量标准化层。在评估时,您需要修复这些问题;否则,在处理每个批次之后,运行方式会不断调整,从而给您不同的精度。在
尝试设置:
评估前。注意:在重新开始训练之前,请致电
model.train()
。在相关问题 更多 >
编程相关推荐