teacher forcing with combining

2026-09-11 12:00:24 +08:00 · 2018-02-23 08:35:53 -08:00
parent c72b8fd64c
commit b5f2181e04
2 changed files with 20 additions and 9 deletions
@@ -34,12 +34,13 @@ class Tacotron(nn.Module):
        encoder_outputs = self.encoder(inputs)

        if self.use_memory_mask:
-            memory_lengths = input_lengths
+            input_lengths = input_lengths
        else:
-            memory_lengths = None
+            input_lengths = None
+
        # (B, T', mel_dim*r)
        mel_outputs, alignments = self.decoder(
-            encoder_outputs, mel_specs, memory_lengths=memory_lengths)
+            encoder_outputs, mel_specs, input_lengths=input_lengths)

        # Post net processing below