From 9504b71f79cd58cad456654aa4d28740662ff3d8 Mon Sep 17 00:00:00 2001 From: erogol Date: Fri, 1 May 2020 23:06:51 +0200 Subject: [PATCH] fix lstm biases True --- layers/tacotron2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/layers/tacotron2.py b/layers/tacotron2.py index 10c03570..4454c89e 100644 --- a/layers/tacotron2.py +++ b/layers/tacotron2.py @@ -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)