Update horovod.py (#11917)

2022-02-16 21:58:54 -08:00 · 2022-02-16 21:58:54 -08:00 · 4dba492fb5
parent d902b59886
commit 4dba492fb5
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/strategies/horovod.py
+++ b/pytorch_lightning/strategies/horovod.py
@ -85,7 +85,7 @@ class HorovodStrategy(ParallelStrategy):
        self._exit_stack = ExitStack()
        self._exit_stack.__enter__()

-        if not self.lightning_module.trainer.training:
+        if not trainer.training:
            # no need to setup optimizers
            return