From 5212a11836a5aa0fca7414cca9803c1beb965fe9 Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Mon, 1 Apr 2019 12:07:24 +0200
Subject: [PATCH] longer stop_token padding

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

diff --git a/layers/tacotron2.py b/layers/tacotron2.py
index b9ee1c5c..70c26743 100644
--- a/layers/tacotron2.py
+++ b/layers/tacotron2.py
@@ -409,7 +409,7 @@ class Decoder(nn.Module):
             stop_flags[2] = t > inputs.shape[1] * 2
             if all(stop_flags):
                 stop_count += 1
-                if stop_count > 10:
+                if stop_count > 20:
                     break
             elif len(outputs) == self.max_decoder_steps:
                 print("   | > Decoder stopped with 'max_decoder_steps")