CURRENT STASH - REVERT THIS

Author: carmocca

pytorch_lightning/loops/epoch/training_epoch_loop.py

@@ -299,7 +299,7 @@ def on_run_end(self) -> None:
             self.update_lr_schedulers("epoch", update_plateau_schedulers=True)
 
             # FIXME: where should this go...
-            self.batch_progress.current.reset()
+            # self.batch_progress.current.reset()
 
         # if fault tolerant is enabled and process has been notified, exit.
         self.trainer._exit_gracefully_on_signal()