From e4e29f716e5bc2049d3316896401fae64933a716 Mon Sep 17 00:00:00 2001 From: Edresson Casanova Date: Mon, 4 May 2020 17:39:35 -0300 Subject: [PATCH] fix bug in bidirectional decoder train --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 94ccfedb..1444e103 100644 --- a/train.py +++ b/train.py @@ -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()})