read

Lightning-AI · Sep 28, 2020 · 0586d5c · 0586d5c
1 parent 3841892
commit 0586d5c
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/pytorch_lightning/accelerators/base_backend.py b/pytorch_lightning/accelerators/base_backend.py
@@ -168,8 +168,9 @@ def early_stopping_should_stop(self, pl_module):
         return self.trainer.should_stop
 
     def setup_optimizers(self, model):
-        if not self.trainer.testing:
-            optimizers, lr_schedulers, optimizer_frequencies = self.trainer.init_optimizers(model)
-            self.trainer.optimizers = optimizers
-            self.trainer.lr_schedulers = lr_schedulers
-            self.trainer.optimizer_frequencies = optimizer_frequencies
+        if self.trainer.testing:
+            return
+        optimizers, lr_schedulers, optimizer_frequencies = self.trainer.init_optimizers(model)
+        self.trainer.optimizers = optimizers
+        self.trainer.lr_schedulers = lr_schedulers
+        self.trainer.optimizer_frequencies = optimizer_frequencies