I'm afraid I just had an improper indentation somewhere, no harms done. Thanks for the reply.

Upon removing the phase loop the epoch was detected automatically again.

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:

LOL, no worries 🙂

Hi TrickyRaccoon92
If you are reporting to tensor-board, then "iteration" equals step. Is this the case?