on_train_epoch_end() runs int he middle of epochs

[inigoval]

Hi
I'm trying to calculate some metrics and generate some images to save at the end of each epoch.
I put this code in the on_train_epoch_end() (I also tried using a custom callback) but the function seems to be called in the middle of the epochs, approximately 3-4 times per epoch.

Surely this isn't intended behaviour? Could it be to do with me using a combined loader?

[edgarriba]

@inigoval thanks for reporting this. Could you provide a snippet code so that we can reproduce ?

[inigoval]

Hi Edgar,
I think that the problem was in the logger that I was using, rather than pytorch lightning and I have now fixed it.
Apologies for reporthing this before investigating it further!

[edgarriba]

@inigoval glad to hear we are bug-free here :)