python - 基于 Loss 和 IoU 如何知道何时停止训练 CNN 模型？

Question

我使用 CNN 模型进行分割任务。输出训练如下：

如何知道我应该何时停止基于损失和 IoU 的训练？

标签： pythontensorflowdeep-learningloss

Answer 1

当损失最小时停止训练。当损失开始增加（验证误差最小）或准确性开始下降时，停止训练。在您的图像中，这似乎是 Epoch 23，迭代 12600。IOU>0.5 被认为是一个很好的预测。