I'm afraid I just had an improper indentation somewhere, no harms done. Thanks for the reply.
Hi TrickyRaccoon92
If you are reporting to tensor-board, then "iteration" equals step. Is this the case?
Interesting... TrickyRaccoon92 could it be the validation phase was creating a new Tensorboard file ?
Hi AgitatedDove14
Turns out my double loop caused some issues.
for e in range(num_epochs): for phase in ['train','valid']: for batch in dataloader:
Upon removing the phase
loop the epoch was detected automatically again.