A big revision: visualization, data loader, tests

2026-09-24 13:10:16 +08:00 · 2018-02-04 08:25:00 -08:00
parent 86f3a56f59
commit 9f5e102473
14 changed files with 199 additions and 64 deletions
@@ -30,7 +30,7 @@ class Tacotron(nn.Module):

        inputs = self.embedding(characters)
        # (B, T', in_dim)
-        encoder_outputs = self.encoder(inputs, input_lengths)
+        encoder_outputs = self.encoder(inputs)

        if self.use_memory_mask:
            memory_lengths = input_lengths