remove redundant fix

Lightning-AI · Aug 17, 2021 · db9acb7 · db9acb7
1 parent d9553da
commit db9acb7
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/pytorch_lightning/loops/epoch/training_epoch_loop.py b/pytorch_lightning/loops/epoch/training_epoch_loop.py
@@ -225,10 +225,6 @@ def on_run_end(self) -> List[List[STEP_OUTPUT]]:
         if self._num_training_batches_reached(self.is_last_batch):
             self.update_lr_schedulers("epoch", update_plateau_schedulers=True)
 
-        self.batch_progress.current.reset()
-        self.scheduler_progress.current.reset()
-        self.batch_loop.optim_progress.reset_on_epoch()
-
         epoch_output = self._epoch_output
         # free memory
         self._epoch_output = None