From 0c5d0b98d8853e966341184a82fb78f9327ee82e Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Mon, 16 Apr 2018 11:54:49 -0700 Subject: [PATCH] threshold changed --- layers/tacotron.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/layers/tacotron.py b/layers/tacotron.py index 0229e9a5..d8956ac0 100644 --- a/layers/tacotron.py +++ b/layers/tacotron.py @@ -323,6 +323,6 @@ class Decoder(nn.Module): return outputs, alignments -def is_end_of_frames(output, alignment, eps=0.05): # 0.2 +def is_end_of_frames(output, alignment, eps=0.01): # 0.2 return ((output.data <= eps).prod(0) > 0).any() \ and alignment.data[:, int(alignment.shape[1]/2):].sum() > 0.7