From 14dbc374973a853e599397aaa16fb5468ab4f00e Mon Sep 17 00:00:00 2001
From: ntt123 <ntt123@earth>
Date: Mon, 10 Feb 2020 16:07:00 +0800
Subject: [PATCH] fix error when batch size = 1

---
 model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model.py b/model.py
index 4c7d7d2..ec0e9ce 100644
--- a/model.py
+++ b/model.py
@@ -407,7 +407,7 @@ class Decoder(nn.Module):
             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(