I used a dataloader to get the batch of some image data.My idea is that I set the max_epoch manually,  Now I use only one image to overfit the network to see it performance.And I use for data in iter(Mydataloader) to get the  batch data.But it seems after one epoch ,the dataloader exhausted and generate nothing, code is still running without warning,and the training loop is not running for a exhausted dataloader.
How to modify?  I want to get data in every epoch.So I need a dataloder to go through the whole dataset in one epoch,after this epoch ,it regenerate an iter  for me.Now I need to re-assign the train_iter=iter(train_loader) before every train loop,because the last one is exhausted.