How could I reset dataloader or count data batch with iter instead of epoch

crossposted: https://stackoverflow.com/questions/60311307/how-does-one-reset-the-dataloader-in-pytorch