From 6c4c1065b0a04296b9db38c35a49d0a63124b6c9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Thu, 9 Sep 2021 09:00:27 +0000
Subject: [PATCH] Fix trainer's scheduler restoring

---
 TTS/trainer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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