diff --git a/pytorch_lightning/loops/fit_loop.py b/pytorch_lightning/loops/fit_loop.py index b36ece18e3..09e7a4d8d8 100644 --- a/pytorch_lightning/loops/fit_loop.py +++ b/pytorch_lightning/loops/fit_loop.py @@ -325,9 +325,6 @@ class FitLoop(Loop[None]): self.trainer._call_lightning_module_hook("on_train_end") self.trainer._call_strategy_hook("on_train_end") - # give accelerators a chance to finish - self.trainer.strategy.on_train_end() - def teardown(self) -> None: if self._data_fetcher is not None: self._data_fetcher.teardown()