diff --git a/layers/tacotron.py b/layers/tacotron.py
index 0e601512..809b78ae 100644
--- a/layers/tacotron.py
+++ b/layers/tacotron.py
@@ -176,7 +176,7 @@ class CBHG(nn.Module):
 
         # (B, T_in, in_features*2)
         # TODO: replace GRU with convolution as in Deep Voice 3
-        self.gru.flatten_parameters()
+        # self.gru.flatten_parameters()
         outputs, _ = self.gru(x)
         return outputs