From d94782a076662f906026572ea2744549e549c7bb Mon Sep 17 00:00:00 2001 From: erogol Date: Mon, 2 Nov 2020 13:18:56 +0100 Subject: [PATCH] reset the way ga_loss is stored in return_dict --- TTS/tts/layers/losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/tts/layers/losses.py b/TTS/tts/layers/losses.py index 50809de5..bafa9440 100644 --- a/TTS/tts/layers/losses.py +++ b/TTS/tts/layers/losses.py @@ -346,7 +346,7 @@ class TacotronLoss(torch.nn.Module): if self.config.ga_alpha > 0: ga_loss = self.criterion_ga(alignments, input_lens, alignment_lens) loss += ga_loss * self.ga_alpha - return_dict['ga_loss'] = ga_loss * self.ga_alpha + return_dict['ga_loss'] = ga_loss # decoder differential spectral loss if self.config.decoder_diff_spec_alpha > 0: