wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

use stop token again

Browse Source
This commit is contained in:
Eren Golge
2019-04-18 15:20:19 +02:00
parent e05263769e
commit f450fe3571
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
layers/tacotron2.py
+2 -2
View File
@@ -445,7 +445,7 @@ class Decoder(nn.Module):
self.attention_layer.init_forward_attn_state(inputs)
outputs, stop_tokens, alignments, t = [], [], [], 0
stop_flags = [True, False, False]
stop_flags = [False, False, False]
stop_count = 0
while True:
memory = self.prenet(memory)
@@ -456,7 +456,7 @@ class Decoder(nn.Module):
alignments += [alignment]
stop_flags[0] = stop_flags[0] or stop_token > 0.5
stop_flags[1] = stop_flags[1] or (alignment[0, -2:].sum() > 0.5 and t > inputs.shape[1])
stop_flags[1] = stop_flags[1] or (alignment[0, -2:].sum() > 0.8 and t > inputs.shape[1])
stop_flags[2] = t > inputs.shape[1] * 2
if all(stop_flags):
stop_count += 1