Update trainer.py

2019-07-26 08:27:14 -04:00 · 2019-07-26 08:27:14 -04:00 · 98fcc17135
parent 5ba0a8fe4c
commit 98fcc17135
1 changed files with 3 additions and 2 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -612,6 +612,7 @@ class Trainer(TrainerIO):
        # run all epochs
        for epoch_nb in range(self.current_epoch, self.max_nb_epochs):
            # update the lr scheduler
+            if self.lr_schedulers is not None:
                for lr_scheduler in self.lr_schedulers:
                    lr_scheduler.step()