Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2026-07-12 21:43:17 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix bug in bidirectional decoder train

Browse Source
This commit is contained in:
Edresson Casanova
2020-05-04 17:39:35 -03:00
committed by GitHub
parent f7b1cad9ee
commit e4e29f716e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@@ -356,7 +356,7 @@ def evaluate(model, criterion, ap, global_step, epoch):
mel_lengths, decoder_backward_output,
alignments, alignment_lengths, text_lengths)
if c.bidirectional_decoder:
keep_avg.update_values({'avg_decoder_b_loss': loss_dict['decoder_backward_loss'].item(),
keep_avg.update_values({'avg_decoder_b_loss': loss_dict['decoder_b_loss'].item(),
'avg_decoder_c_loss': loss_dict['decoder_c_loss'].item()})
if c.ga_alpha > 0:
keep_avg.update_values({'avg_ga_loss': loss_dict['ga_loss'].item()})