Merge branch 'dev' of https://github.com/mozilla/TTS into dev

This commit is contained in:
Eren Gölge committed 2020-05-18 13:12:31 +02:00
commit e55e28bb37
5 files changed
+10 -5

No files matched your search

+1 -1
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()})