wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

checkpoint log

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
utils/generic_utils.py
+1 -1
View File
@@ -113,7 +113,7 @@ def save_checkpoint(model, optimizer, optimizer_st, model_loss, out_path,
current_step, epoch):
checkpoint_path = 'checkpoint_{}.pth.tar'.format(current_step)
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()
state = {