diff --git a/TTS/trainer.py b/TTS/trainer.py index 13439861..e5867ac9 100644 --- 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