Hi @pytorchnewbie

The epoch_num that is printed is always 0. That’s not right since it prints every epoch - why would this happen?

This a correct behaviour as passing trainer.state.epoch it is passed as value and not reference. I agree that it would be nice to have it as reference, but I have no idea how to do that in python.
You need to fetch epoch inside the handler with trainer.state.epoch.