mirror of
https://github.com/wassname/TTS.git
synced 2026-10-26 12:20:32 +08:00
fix lstm biases True
This commit is contained in:
1 parent
de2918c85b
commit
9504b71f79
1 file changed
+2
-2
+2
-2
@@ -123,7 +123,7 @@ class Decoder(nn.Module):
|
||||
|
||||
self.attention_rnn = nn.LSTMCell(self.prenet_dim + input_dim,
|
||||
self.query_dim,
|
||||
bias=False)
|
||||
bias=True)
|
||||
|
||||
self.attention = init_attn(attn_type=attn_type,
|
||||
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_dim,
|
||||
bias=False)
|
||||
bias=True)
|
||||
|
||||
self.linear_projection = Linear(self.decoder_rnn_dim + input_dim,
|
||||
self.frame_dim * self.r_init)
|
||||
|
||||
Reference in new issue
Block a user