Hi, I want to load last saved batch id and batch data from Dataloader.
batch_id = checkpoint['batch_id'] #value from last iteration
batch_data = checkpoint['batch_data'] #value from last iteration

for batch_id, batch_data in enumerate(train_iter): #want to load train_iter with above values