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

some more tests for model saving

This commit is contained in:
Eren Golge 2018-02-21 07:21:44 -08:00
parent 21a4ee63fe
commit 9b7b5e238e
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -218,8 +218,7 @@ def main(args):
if c.checkpoint: if c.checkpoint:
# save model # save model
save_checkpoint(model, optimizer, linear_loss.data[0], save_checkpoint(model, optimizer, linear_loss.data[0],
best_loss, OUT_PATH, OUT_PATH, current_step, epoch)
current_step, epoch)
# Diagnostic visualizations # Diagnostic visualizations
const_spec = linear_output[0].data.cpu().numpy() const_spec = linear_output[0].data.cpu().numpy()

2
utils/generic_utils.py
View File

@ -60,7 +60,7 @@ def _trim_model_state_dict(state_dict):
return new_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): 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)