Why is there no training_epoch_end?
🚀 Feature
If i want to calculate and log average statistics for the training epoch, it seems like there is no option to define a "training_epoch_end" in the LightningModule, as there is validation_epoch_end and test_epoch_end.
Motivation
Seems very intuitive to have this function. I know the on_epoch_end hook exists, but the "outputs" object with training history for that epoch is not available.
Pitch
Same behavior of validation_epoch_end and test_epoch_end in training.
Sorry if there is something like this already, just started to use Pl. (the master version).