updates and debugs

2026-09-09 11:16:00 +08:00 · 2018-02-13 01:45:52 -08:00
parent dadefb5dbc
commit 56697ac8cf
10 changed files with 121 additions and 63 deletions
@@ -20,7 +20,7 @@ class Tacotron(nn.Module):
        # Trying smaller std
        self.embedding.weight.data.normal_(0, 0.3)
        self.encoder = Encoder(embedding_dim)
-        self.decoder = Decoder(mel_dim, r)
+        self.decoder = Decoder(256, mel_dim, r)

        self.postnet = CBHG(mel_dim, K=8, projections=[256, mel_dim])
        self.last_linear = nn.Linear(mel_dim * 2, freq_dim)