tacotron2 stop condition

2020-02-03 15:29:44 +01:00 · 2020-02-03 15:29:44 +01:00 · 1ef6278d2d
parent ffe9a32813
commit 1ef6278d2d
1 changed files with 1 additions and 1 deletions
--- a/layers/tacotron2.py
+++ b/layers/tacotron2.py
@ -290,7 +290,7 @@ class Decoder(nn.Module):
            stop_tokens += [stop_token]
            alignments += [alignment]

-            if stop_token > 0.7:
+            if stop_token > 0.7 and t > inputs.shape[0] / 2:
                break
            if len(outputs) == self.max_decoder_steps:
                print("   | > Decoder stopped with 'max_decoder_steps")