mirror of
https://github.com/wassname/TTS.git
synced 2026-10-26 12:20:32 +08:00
Remove flatten_parameters due to a bug at pytorch 0.4
This commit is contained in:
1 parent
9ff240e0ea
commit
6bcec24d13
1 file changed
+1
-1
+1
-1
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user