mirror of https://github.com/coqui-ai/TTS.git
threshold changed
This commit is contained in:
parent
e00bc6693b
commit
0c5d0b98d8
|
@ -323,6 +323,6 @@ class Decoder(nn.Module):
|
||||||
return outputs, alignments
|
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() \
|
return ((output.data <= eps).prod(0) > 0).any() \
|
||||||
and alignment.data[:, int(alignment.shape[1]/2):].sum() > 0.7
|
and alignment.data[:, int(alignment.shape[1]/2):].sum() > 0.7
|
||||||
|
|
Loading…
Reference in New Issue