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

checkpoint log

This commit is contained in:
erogol 2020-04-23 14:24:38 +02:00
parent 0e7ecca33f
commit d5093bf6fb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/generic_utils.py
View File

@ -113,7 +113,7 @@ def save_checkpoint(model, optimizer, optimizer_st, model_loss, out_path,
current_step, epoch): current_step, epoch):
checkpoint_path = 'checkpoint_{}.pth.tar'.format(current_step) checkpoint_path = 'checkpoint_{}.pth.tar'.format(current_step)
checkpoint_path = os.path.join(out_path, checkpoint_path) checkpoint_path = os.path.join(out_path, checkpoint_path)
print(" | | > Checkpoint saving : {}".format(checkpoint_path)) print(" > CHECKPOINT : {}".format(checkpoint_path))
new_state_dict = model.state_dict() new_state_dict = model.state_dict()
state = { state = {