From 4e93f90108cfcafeac7f9a4b456510e39e9af71f Mon Sep 17 00:00:00 2001 From: erogol Date: Tue, 22 Sep 2020 15:15:40 +0200 Subject: [PATCH] bug fix --- 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 d7d45c1b..79654c07 100644 --- a/TTS/tts/layers/losses.py +++ b/TTS/tts/layers/losses.py @@ -220,7 +220,7 @@ class TacotronLoss(torch.nn.Module): if self.decoder_alpha > 0: decoder_loss = self.criterion(decoder_output, mel_input, output_lens) - if postnet_alpha > 0: + if self.postnet_alpha > 0: if self.config.model in ["Tacotron", "TacotronGST"]: postnet_loss = self.criterion(postnet_output, linear_input, output_lens)