Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove flatten_parameters due to a bug at pytorch 0.4

This commit is contained in:
Eren Golge 2018-05-18 06:00:16 -07:00
parent 9ff240e0ea
commit 6bcec24d13
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
layers/tacotron.py
View File

@ -176,7 +176,7 @@ class CBHG(nn.Module):
# (B, T_in, in_features*2) # (B, T_in, in_features*2)
# TODO: replace GRU with convolution as in Deep Voice 3 # TODO: replace GRU with convolution as in Deep Voice 3
self.gru.flatten_parameters() # self.gru.flatten_parameters()
outputs, _ = self.gru(x) outputs, _ = self.gru(x)
return outputs return outputs