pytorch-image-models
pytorch-image-models copied to clipboard
[FEATURE] Save checkpoint interval
How can I edit the save-checkpoint inverval for my model?
While training with high epochs, old checkpoints are often overwitten by new checkpoints. In my case, the last 10 checkpoints are usually over-fit so I want to avoid that by adding saving interval, keeping my old checkpoints without taking up too much spaces for checkpointing. How can I do this?
Thank you!
@ngfuong there's no interval to skip some subset of epochs for eval & save ... but you can use --checkpoint-history 20
to say keep the last 20 epochs of eval checkpoints.
@ngfuong there's no interval to skip some subset of epochs for eval & save ... but you can use
--checkpoint-history 20
to say keep the last 20 epochs of eval checkpoints.
Thank you for replying! Since I intend to implement this feature, which files should I modify to add this?