mirror of https://github.com/coqui-ai/TTS.git
linter fix
This commit is contained in:
parent
7e669b38a9
commit
32e8b56c45
|
@ -313,7 +313,7 @@ class TacotronLoss(torch.nn.Module):
|
||||||
output_lens)
|
output_lens)
|
||||||
if self.postnet_alpha > 0:
|
if self.postnet_alpha > 0:
|
||||||
postnet_loss = self.criterion(postnet_output, postnet_target,
|
postnet_loss = self.criterion(postnet_output, postnet_target,
|
||||||
output_lens)
|
output_lens)
|
||||||
else:
|
else:
|
||||||
if self.decoder_alpha > 0:
|
if self.decoder_alpha > 0:
|
||||||
decoder_loss = self.criterion(decoder_output, mel_input)
|
decoder_loss = self.criterion(decoder_output, mel_input)
|
||||||
|
|
Loading…
Reference in New Issue