From 443659a7317870d15acd4579be9f63fa92053fcf Mon Sep 17 00:00:00 2001 From: Eren Date: Thu, 6 Sep 2018 15:31:07 +0200 Subject: [PATCH] reverse post-cbhg size --- layers/tacotron.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/layers/tacotron.py b/layers/tacotron.py index 336f2d9e..b0327917 100644 --- 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)