search

IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
Apache License 2.0
2.1k stars 230 forks source link

Which epoch run produces the best result? #195

Open htcml opened 1 year ago

htcml commented 1 year ago

Say, for example, set epoch = 24. After training, how do I know which epoch run produces the best checkpoint?

checkpoint0023_4scale.pth obviously was produced by epoch=24 training. But how did you know 23th epoch run produced the best model?

half-truism commented 1 year ago

In main.py they track the best ckpt with checkpoint_best_regular.pth during training, which is tracked based off the best map value. If you want to you could change the name of the file at line 309 to include the epoch number https://github.com/IDEA-Research/DINO/blob/main/main.py#L309