Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2026-07-13 05:50:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

bug fix for TTS model restore

Browse Source
This commit is contained in:
erogol
2020-06-06 12:59:59 +02:00
parent d2cfab71d3
commit b736c6cdac
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@@ -542,7 +542,7 @@ def main(args): # pylint: disable=redefined-outer-name
except:
print(" > Partial model initialization.")
model_dict = model.state_dict()
model_dict = set_init_dict(model_dict, checkpoint, c)
model_dict = set_init_dict(model_dict, checkpoint['model'], c)
model.load_state_dict(model_dict)
del model_dict
for group in optimizer.param_groups: