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 · 6bcec24d13
parent 9ff240e0ea
commit 6bcec24d13
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