Also update progress_bar in training_epoch_end

[Deanamic]

🚀 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

[github-actions]

Hi! thanks for your contribution!, great first issue!

[awaelchli]

I will look into this after #1450 is done :)