From 9ac636335eb4f5e9d16141dcbe3ad84796da4d3d Mon Sep 17 00:00:00 2001 From: ananthsub Date: Fri, 25 Mar 2022 03:38:09 -0700 Subject: [PATCH] Update fit_loop.py (#12450) --- pytorch_lightning/loops/fit_loop.py | 3 --- 1 file changed, 3 deletions(-) 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()