So, in the training loop I can do something l can loop for each epoch, and instead of enumerating on the dataloader, just use  data, target = next(iter(dataloader))  and then do my calls to model, optimizer, and loss?