-
同一张图片,模型每次预测的结果都不一样,准确率还莫名其妙地掉了:我忘了在 PyTorch 推理前调用 model.eval() 的复盘
训练好的图像分类模型拿去推理,同一张图预测两次结果竟然不一样,整体准确率还比验证集低一截。检查输入和权重都没问题,最后发现是我推理前忘了调 model.eval()——模型停在训练模式,Dropout 还在随机丢弃神经元(导致结果随机)、BatchNorm 还用当前 batch 统计量(导致单张图变差)。这篇从 Dropout/BatchNorm 训练推理为何不同讲到 model.eval()+t…- 0
- 0
Dropout
-
￥优惠劵使用时效：无法使用使用时效：
之前
使用时效：永久有效优惠劵ID：×