How to save model checkpoints every 1000 batches of data during training

I was tryna to save model multiple checkpoints during training.

For example, after every 1000 batches of data, I want to save one checkpoint for that.

I am not quite sure how to achieve that after I have googled and looked at the doc.

Appreciate any help!

check this: How can I save a model's checkpoint every N optimization steps? · Issue #5473 · PyTorchLightning/pytorch-lightning · GitHub

1 Like

thank you for the help!!!