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

change alignment_score to alignment_error

This commit is contained in:
erogol 2020-06-09 23:01:34 +02:00
parent 9913724a70
commit 0245a2e258
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -291,7 +291,7 @@ def train(model, criterion, optimizer, optimizer_st, scheduler,
"loss_postnet": keep_avg['avg_postnet_loss'], "loss_postnet": keep_avg['avg_postnet_loss'],
"loss_decoder": keep_avg['avg_decoder_loss'], "loss_decoder": keep_avg['avg_decoder_loss'],
"stopnet_loss": keep_avg['avg_stopnet_loss'], "stopnet_loss": keep_avg['avg_stopnet_loss'],
"alignment_score": keep_avg['avg_align_error'], "alignment_error": keep_avg['avg_align_error'],
"epoch_time": epoch_time "epoch_time": epoch_time
} }
if c.ga_alpha > 0: if c.ga_alpha > 0:
@ -413,7 +413,7 @@ def evaluate(model, criterion, ap, global_step, epoch):
"loss_postnet": keep_avg['avg_postnet_loss'], "loss_postnet": keep_avg['avg_postnet_loss'],
"loss_decoder": keep_avg['avg_decoder_loss'], "loss_decoder": keep_avg['avg_decoder_loss'],
"stopnet_loss": keep_avg['avg_stopnet_loss'], "stopnet_loss": keep_avg['avg_stopnet_loss'],
"alignment_score": keep_avg['avg_align_error'], "alignment_error": keep_avg['avg_align_error'],
} }
if c.bidirectional_decoder: if c.bidirectional_decoder: