diff --git a/layers/tacotron.py b/layers/tacotron.py
index b0327917..336f2d9e 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=80,
-            conv_projections=[160, mel_dim],
-            highway_features=80,
-            gru_features=80,
+            conv_bank_features=128,
+            conv_projections=[256, mel_dim],
+            highway_features=128,
+            gru_features=128,
             num_highways=4)
     def forward(self, x):
         return self.cbhg(x)
diff --git a/train.py b/train.py
index c45cfd9d..e5fdc12f 100644
--- a/train.py
+++ b/train.py
@@ -401,7 +401,7 @@ def main(args):
             criterion.cuda()
             criterion_st.cuda()
         optimizer.load_state_dict(checkpoint['optimizer'])
-        optimizer_st.load_state_dict(checkpoint['optimizer_st'])
+        # optimizer_st.load_state_dict(checkpoint['optimizer_st'])
         for state in optimizer.state.values():
             for k, v in state.items():
                 if torch.is_tensor(v):