mirror of https://github.com/coqui-ai/TTS.git
bug fix
This commit is contained in:
parent
bb9b70ee27
commit
4e93f90108
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue