reverse post-cbhg size

2018-09-06 15:31:07 +02:00 · 2018-09-06 15:31:07 +02:00 · 443659a731
parent 56c6d0cac8
commit 443659a731
1 changed files with 4 additions and 4 deletions
--- a/layers/tacotron.py
+++ b/layers/tacotron.py
@ -249,10 +249,10 @@ class PostCBHG(nn.Module):
        self.cbhg = CBHG(
            mel_dim,
            K=8,
-            conv_bank_features=128,
-            conv_projections=[256, mel_dim],
-            highway_features=128,
-            gru_features=128,
+            conv_bank_features=80,
+            conv_projections=[160, mel_dim],
+            highway_features=80,
+            gru_features=80,
            num_highways=4)
    def forward(self, x):
        return self.cbhg(x)