Update fit_loop.py (#12450)

2022-03-25 03:38:09 -07:00 · 2022-03-25 03:38:09 -07:00 · 9ac636335e
parent 812c2dc3d3
commit 9ac636335e
1 changed files with 0 additions and 3 deletions
--- 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()