tacotron2 stop condition

2020-02-03 15:29:44 +01:00 · 2020-02-03 15:29:44 +01:00 · 9feec72d44
parent 542141c978
commit 9feec72d44
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")