longer stop_token padding

2019-04-01 12:07:24 +02:00 · 2019-04-01 12:07:24 +02:00 · 5212a11836
parent e1cd253d65
commit 5212a11836
1 changed files with 1 additions and 1 deletions
--- 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")