mirror of
https://github.com/wassname/TTS.git
synced 2026-10-27 12:30:27 +08:00
bug fix
This commit is contained in:
1 parent
bb9b70ee27
commit
4e93f90108
1 file changed
+1
-1
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user