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 NameError: name 'model_dict' is not defined 

Closes #91
This commit is contained in:
Yves-Noel Weweler 2018-12-31 13:29:39 +01:00
parent e78911bf88
commit 10524b885c
No known key found for this signature in database
GPG Key ID: 44C0C9435C58B88D
1 changed files with 16 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -400,7 +400,10 @@ def main(args):
if args.restore_path: if args.restore_path:
checkpoint = torch.load(args.restore_path) checkpoint = torch.load(args.restore_path)
try:
model.load_state_dict(checkpoint['model']) model.load_state_dict(checkpoint['model'])
except:
model_dict = model.state_dict()
# Partial initialization: if there is a mismatch with new and old layer, it is skipped. # Partial initialization: if there is a mismatch with new and old layer, it is skipped.
# 1. filter out unnecessary keys # 1. filter out unnecessary keys
pretrained_dict = { pretrained_dict = {