Fix trainer's scheduler restoring

2021-09-09 09:00:27 +00:00 · 2021-09-09 09:00:27 +00:00 · 6c4c1065b0
parent 807f1d3817
commit 6c4c1065b0
1 changed files with 2 additions and 1 deletions
--- a/TTS/trainer.py
+++ b/TTS/trainer.py
@ -275,7 +275,8 @@ class Trainer:
            if self.args.continue_path:
                if isinstance(self.scheduler, list):
                    for scheduler in self.scheduler:
-                        scheduler.last_epoch = self.restore_step
+                        if scheduler is not None:
+                            scheduler.last_epoch = self.restore_step
                else:
                    self.scheduler.last_epoch = self.restore_step