Pythorch预训练RESNET18 Mod

2024-05-01 21:59:38 发布

您现在位置:Python中文网/ 问答频道 /正文

我在Pythorch中训练了一个预先训练过的RESNET18模型并保存了它。在测试过程中,对于不同的小批量,模型给出了不同的精度。有人知道为什么吗?在


Tags: 模型过程精度pythorch小批量resnet18
1条回答
网友
1楼 · 发布于 2024-05-01 21:59:38

是的,我想是的。 RESNET包含批量标准化层。在评估时,您需要修复这些问题;否则,在处理每个批次之后,运行方式会不断调整,从而给您不同的精度。在

尝试设置:

model.eval()

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

相关问题 更多 >