BDL icon indicating copy to clipboard operation
BDL copied to clipboard

How do you decide when to stop the training ?

Open idealwei opened this issue 5 years ago • 4 comments

Hi, I am curious how to decide when to stop the training and how to choose the final snapshots. It's not clarified in your paper. I found the "Early Stopping" parameters in your code, how to set this hyper-parameter?

idealwei avatar Jul 27 '19 06:07 idealwei

It is a little hard to decide. For training without using SSL, I find stopping at 80000 iterations is best. When I continue to train with more iterations, overfitting will be caused. For training with SSL, I find there is not overfitting, but after 120000 iterations the mIoU starts to be stable. So I choose to stop at 120000 iterations for SSL.

liyunsheng13 avatar Jul 27 '19 22:07 liyunsheng13

In my opinion, when training without SSL, you tested every snapshot on 'val' split of Cityscapes and set the best snapshot iteration as the "Early Stopping" hyper-parameter.

idealwei avatar Jul 28 '19 14:07 idealwei

Not really. The best result is always shown when iteration is around 80000. It is unnecessary to validate all snapshots.

liyunsheng13 avatar Aug 27 '19 22:08 liyunsheng13

But if you have not got the performance of 75000 , 85000, 90000, how did you know that the best results is shown around 80000.

idealwei avatar Aug 31 '19 01:08 idealwei