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

some more tests for model saving

Browse Source
This commit is contained in:
Eren Golge
2018-02-21 07:21:44 -08:00
parent 6efb761139
commit d510f0e8aa
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/generic_utils.py
+1 -1
View File
@@ -60,7 +60,7 @@ def _trim_model_state_dict(state_dict):
return new_state_dict
def save_checkpoint(model, optimizer, model_loss, best_loss, out_path,
def save_checkpoint(model, optimizer, model_loss, out_path,
current_step, epoch):
checkpoint_path = 'checkpoint_{}.pth.tar'.format(current_step)
checkpoint_path = os.path.join(out_path, checkpoint_path)