threshold changed

pull/10/head
Eren Golge 2018-04-16 11:54:49 -07:00
parent e00bc6693b
commit 0c5d0b98d8
1 changed files with 1 additions and 1 deletions

View File

@ -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