How to resume training from batch index

[anjul1008]

Hi, I saw new step the checkpoint saving feature recently. I noticed training starts from batch index 0, how to resume training from the same batch index where i stopped the training?

I didn't find the direct way as it happens in pytorch dataloader. Can you provide some suggestion?

thank you.