pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard

[FEATURE] Save checkpoint interval

Open ngfuong opened this issue 2 years ago • 1 comments

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 avatar Aug 01 '22 03:08 ngfuong

@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.

rwightman avatar Aug 08 '22 20:08 rwightman

@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?

ngfuong avatar Aug 17 '22 10:08 ngfuong