Epoch end hook in callback

Hi!

I created a callback that should aggregate metrics after train and validation epochs. For this, I use the on_epoch_end hook. But it called between train and val. So, am I wrong in understanding the order of hooks? Which hook I should use in a callback to handle metrics after each training and validating cycle?

you can try on_validation_epoch_end or on_validation_end hook

1 Like