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

if reinit option is used run runtime rrror to partially iniot the model

This commit is contained in:
Eren Golge 2019-03-29 17:03:29 +01:00
parent 6edd8bc6dd
commit e1573440ba
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -392,6 +392,8 @@ def main(args):
# TODO: fix optimizer init, model.cuda() needs to be called before # TODO: fix optimizer init, model.cuda() needs to be called before
# optimizer restore # optimizer restore
# optimizer.load_state_dict(checkpoint['optimizer']) # optimizer.load_state_dict(checkpoint['optimizer'])
if len(c.reinit_layers) > 0:
raise RuntimeError
model.load_state_dict(checkpoint['model']) model.load_state_dict(checkpoint['model'])
except: except:
print(" > Partial model initialization.") print(" > Partial model initialization.")