From afbfb6f1a65251ca0337911d2ffd6866aa861a35 Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Fri, 18 May 2018 06:00:16 -0700
Subject: [PATCH] Remove flatten_parameters due to a bug at pytorch 0.4

---
 layers/tacotron.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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