bug fixes

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)