Search code examples
pythontensorflowdeep-learningloss

how to know when stop training in CNN model based on Loss and IoU?


I used a CNN model for segmentation task. the output training is like below :

enter image description here

how to know when should I stop training based on Loss and IoU?

enter image description here

enter image description here


Solution

  • Stop training when the loss is at minimum. When loss begins to increase(validation error is at its minimum) or accuracy begins to decrease, stop the training. In your image this appears to be Epoch 23, iteration 12600. IOU>0.5 is considered a good prediction.