linter fix

This commit is contained in:
Eren Gölge
2021-03-11 18:47:16 +01:00
parent ddab50b515
commit bf0caba0bc

View File

@@ -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)