yassouali/CCT

saving best model

SuzannaLin opened this issue · 2 comments

Hi Yassine,
During training, is always the latest model saved or only the best?
If the best model is saved, on which metric is that decision based?
For example: This is what I got after 76 epochs
image

but at 78 the model stopped and its mIoU for the EVAL set is lower:
image

Hi @SuzannaLin the best model is saved based on mIoU, this is specified in the config file "monitor": "max Mean_IoU"

Thanks for your answer!