From cac7e9ca5b9fef757a3ce62f814a0bf8bc8890b7 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Sun, 13 May 2018 06:31:59 -0700 Subject: [PATCH] Stop test time model with stop_token --- layers/tacotron.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/layers/tacotron.py b/layers/tacotron.py index 1304fa74..7df5b1e8 100644 --- a/layers/tacotron.py +++ b/layers/tacotron.py @@ -309,7 +309,7 @@ class Decoder(nn.Module): if t >= T_decoder: break else: - if t > 1 and is_end_of_frames(output.view(self.r, -1), alignment, self.eps): + if t > 1 and stop_token > 0.5: break elif t > self.max_decoder_steps: print(" !! Decoder stopped with 'max_decoder_steps'. \