diff --git a/TTS/tts/layers/losses.py b/TTS/tts/layers/losses.py index c5497054..213970a7 100644 --- a/TTS/tts/layers/losses.py +++ b/TTS/tts/layers/losses.py @@ -313,7 +313,7 @@ class TacotronLoss(torch.nn.Module): output_lens) if self.postnet_alpha > 0: postnet_loss = self.criterion(postnet_output, postnet_target, - output_lens) + output_lens) else: if self.decoder_alpha > 0: decoder_loss = self.criterion(decoder_output, mel_input)