mirror of https://github.com/coqui-ai/TTS.git
linter fix
This commit is contained in:
parent
edc5b07d75
commit
d652fb4999
|
@ -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