Pytorch_Retinaface icon indicating copy to clipboard operation
Pytorch_Retinaface copied to clipboard

How to find the best model when training

Open quocnhat opened this issue 5 years ago • 2 comments
trafficstars

Hi, Thanks very much for sharing your awesome works. Due to the variation of the dataset ( hard & easy examples in each iteration), the out loss is unstable at all. So how can we determine the best model when training?

quocnhat avatar Aug 11 '20 02:08 quocnhat

Same problem. I wonder how to calculate AP during training in order to find the best model. The provided evaluation method seems cannot be used in the training.

DerekChen88 avatar Aug 13 '20 07:08 DerekChen88

for now, The simplest method is to save the best loss checkpoint crosses the number of epochs. I will evaluate that method when finish.

quocnhat avatar Aug 13 '20 08:08 quocnhat