From 772c859dffa2fe60f0a4d2fb7778be5344646e84 Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Thu, 7 Mar 2019 11:44:17 +0100
Subject: [PATCH] Chnage stop condition

---
 layers/tacotron2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/layers/tacotron2.py b/layers/tacotron2.py
index 296ea7ec..e743186f 100644
--- a/layers/tacotron2.py
+++ b/layers/tacotron2.py
@@ -355,7 +355,7 @@ class Decoder(nn.Module):
             alignments += [alignment]
 
             stop_flags[0] = stop_flags[0] or gate_output > 0.5
-            stop_flags[1] = stop_flags[1] or alignment[0, -3:].sum() > 0.5
+            stop_flags[1] = stop_flags[1] or alignment[0, -2:].sum() > 0.5
             if all(stop_flags):
                 break
             elif len(outputs) == self.max_decoder_steps: