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

fix lstm biases True

This commit is contained in:
erogol 2020-05-01 23:06:51 +02:00
parent de2918c85b
commit 9504b71f79
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
layers/tacotron2.py
View File

@ -123,7 +123,7 @@ class Decoder(nn.Module):
self.attention_rnn = nn.LSTMCell(self.prenet_dim + input_dim, self.attention_rnn = nn.LSTMCell(self.prenet_dim + input_dim,
self.query_dim, self.query_dim,
bias=False) bias=True)
self.attention = init_attn(attn_type=attn_type, self.attention = init_attn(attn_type=attn_type,
query_dim=self.query_dim, query_dim=self.query_dim,
@ -141,7 +141,7 @@ class Decoder(nn.Module):
self.decoder_rnn = nn.LSTMCell(self.query_dim + input_dim, self.decoder_rnn = nn.LSTMCell(self.query_dim + input_dim,
self.decoder_rnn_dim, self.decoder_rnn_dim,
bias=False) bias=True)
self.linear_projection = Linear(self.decoder_rnn_dim + input_dim, self.linear_projection = Linear(self.decoder_rnn_dim + input_dim,
self.frame_dim * self.r_init) self.frame_dim * self.r_init)