Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix trainer's scheduler restoring

This commit is contained in:
Eren Gölge 2021-09-09 09:00:27 +00:00
parent 807f1d3817
commit 6c4c1065b0
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
TTS/trainer.py
View File

@ -275,6 +275,7 @@ class Trainer:
if self.args.continue_path: if self.args.continue_path:
if isinstance(self.scheduler, list): if isinstance(self.scheduler, list):
for scheduler in self.scheduler: for scheduler in self.scheduler:
if scheduler is not None:
scheduler.last_epoch = self.restore_step scheduler.last_epoch = self.restore_step
else: else:
self.scheduler.last_epoch = self.restore_step self.scheduler.last_epoch = self.restore_step