From de2918c85b5afb2648d2f39a0a47fcee204ba101 Mon Sep 17 00:00:00 2001 From: erogol Date: Fri, 1 May 2020 14:34:14 +0200 Subject: [PATCH] bug fixes --- layers/tacotron2.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/layers/tacotron2.py b/layers/tacotron2.py index 35a5c0bb..10c03570 100644 --- a/layers/tacotron2.py +++ b/layers/tacotron2.py @@ -82,8 +82,8 @@ class Encoder(nn.Module): o = x for layer in self.convolutions: o = layer(o) - o = x.transpose(1, 2) - self.lstm.flatten_parameters() + o = o.transpose(1, 2) + # self.lstm.flatten_parameters() o, _ = self.lstm(o) return o @@ -140,7 +140,8 @@ class Decoder(nn.Module): attn_K=attn_K) self.decoder_rnn = nn.LSTMCell(self.query_dim + input_dim, - self.decoder_rnn_dim, 1) + self.decoder_rnn_dim, + bias=False) self.linear_projection = Linear(self.decoder_rnn_dim + input_dim, self.frame_dim * self.r_init)