Tensorflow Object Detection - mAP 稳定，但 Total Loss 正在下降。继续训练？

Posted 2023-03-27

【中文标题】Tensorflow Object Detection - mAP 稳定，但 Total Loss 正在下降。继续训练？

【英文标题】：Tensorflow Object Detection - mAP steady, but Total Loss is dropping. Keep training?

【发布时间】：2019-11-05 19:42:00

【问题描述】：

使用 Tensorflow 1.12.2 和 Python 3.6 训练 Object Detection 分类器。使用 4 个类别的大约 20,000 张图像，经过 160K 步后，mAP 相当稳定，但总损失仍在下降（尽管非常缓慢）。我应该继续训练直到总损失停止下降或上升吗？

【参考方案1】：

当 Total Loss 向上时，你很可能已经进入了过度拟合。

一个好主意是对照测试数据集检查您当前的快照，看看您是否得到满意的结果。如果结果好，我会停止训练。