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

Add the ValueError in the restore checkpoint exception to avoid problems with the optimizer restauration when new keys are addition

This commit is contained in:
Edresson 2021-08-28 10:11:33 -03:00 committed by Eren Gölge
parent bbdb5c38e6
commit 0bdfd3cb50
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/trainer.py
View File

@ -453,7 +453,7 @@ class Trainer:
if "scaler" in checkpoint and self.use_amp_scaler and checkpoint["scaler"]: if "scaler" in checkpoint and self.use_amp_scaler and checkpoint["scaler"]:
print(" > Restoring Scaler...") print(" > Restoring Scaler...")
scaler = _restore_list_objs(checkpoint["scaler"], scaler) scaler = _restore_list_objs(checkpoint["scaler"], scaler)
except (KeyError, RuntimeError): except (KeyError, RuntimeError, ValueError):
print(" > Partial model initialization...") print(" > Partial model initialization...")
model_dict = model.state_dict() model_dict = model.state_dict()
model_dict = set_init_dict(model_dict, checkpoint["model"], config) model_dict = set_init_dict(model_dict, checkpoint["model"], config)