fix error when batch size = 1

tuong-olli · Feb 10, 2020 · 14dbc37 · 14dbc37
1 parent 6d0635e
commit 14dbc37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -407,7 +407,7 @@ def forward(self, memory, decoder_inputs, memory_lengths):
             mel_output, gate_output, attention_weights = self.decode(
                 decoder_input)
             mel_outputs += [mel_output.squeeze(1)]
-            gate_outputs += [gate_output.squeeze()]
+            gate_outputs += [gate_output.squeeze(1)]
             alignments += [attention_weights]
 
         mel_outputs, gate_outputs, alignments = self.parse_decoder_outputs(