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

bug fix

This commit is contained in:
erogol 2020-06-11 10:48:20 +02:00
parent e8a59db839
commit b8766c05ce
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -237,7 +237,7 @@ def train(model, criterion, optimizer, optimizer_st, scheduler,
# save model # save model
save_checkpoint(model, optimizer, global_step, epoch, model.decoder.r, OUT_PATH, save_checkpoint(model, optimizer, global_step, epoch, model.decoder.r, OUT_PATH,
optimizer_st=optimizer_st, optimizer_st=optimizer_st,
model_loss=loss_dict['postnet_loss'].item()) model_loss=loss_dict['postnet_loss'])
# Diagnostic visualizations # Diagnostic visualizations
const_spec = postnet_output[0].data.cpu().numpy() const_spec = postnet_output[0].data.cpu().numpy()