Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

longer stop_token padding

This commit is contained in:
Eren Golge 2019-04-01 12:07:24 +02:00
parent e1cd253d65
commit 5212a11836
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
layers/tacotron2.py
View File

@ -409,7 +409,7 @@ class Decoder(nn.Module):
stop_flags[2] = t > inputs.shape[1] * 2 stop_flags[2] = t > inputs.shape[1] * 2
if all(stop_flags): if all(stop_flags):
stop_count += 1 stop_count += 1
if stop_count > 10: if stop_count > 20:
break break
elif len(outputs) == self.max_decoder_steps: elif len(outputs) == self.max_decoder_steps:
print(" | > Decoder stopped with 'max_decoder_steps") print(" | > Decoder stopped with 'max_decoder_steps")