Remove flatten_parameters due to a bug at pytorch 0.4

2018-05-18 06:00:16 -07:00 · 2018-05-18 06:00:16 -07:00 · afbfb6f1a6
parent a8a601dd64
commit afbfb6f1a6
1 changed files with 1 additions and 1 deletions
--- 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