RangiLyu / nanodet

NanoDet-Plusāš”Super fast and lightweight anchor-free object detection model. šŸ”„Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphonešŸ”„
Apache License 2.0
5.77k stars 1.04k forks source link

How to save after every epoch on Training? #568

Closed gabxap closed 5 months ago

gabxap commented 5 months ago

Hello, I'd like to know if there's an option within the config file to save the model_best folder (specifically the nanodet_model_best.pth file) after each epoch during my training process in different locations. From what I see after I train I can only get the last epoch model. I'm interested in testing the performance of my model at every epoch.