fix lstm biases True

2026-09-09 11:16:00 +08:00 · 2020-05-12 16:23:32 +02:00
parent de2918c85b
commit 9504b71f79
1 changed files with 2 additions and 2 deletions
@@ -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)