Hey @bavshehata
The hook is actually called on_train_epoch_end
. If you rename it in your code, it will get called. I see that we got that wrong in our release notes, I will fix that. If you got this info from somewhere else, let me know and I can double check that it is correct.