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: