Remove flatten_parameters due to a bug at pytorch 0.4

2026-09-09 11:16:00 +08:00 · 2018-05-18 06:00:16 -07:00
parent 9ff240e0ea
commit 6bcec24d13
1 changed files with 1 additions and 1 deletions
@@ -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