Also update progress_bar in training_epoch_end
🚀 Feature
PR 1357 implements training_epoch_end to log metrics. The comments in the issue suggest that it should behave like validation_epoch_end, but the PR only replicates the callbacks and metric logging. This feature would add updates to the progress bar as well.
Motivation
Motivation is for the same usecase as having a progress bar.
While tensorboard or any other logger is a great resource to have. It is painful to set up in environments like slurm where you can only submit jobs and possibly cannot forward ports or set up the logging.
Pitch
The changes should only entail adding an update to tqdm here