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

Print `max_decoder_steps` when model reaches the limit

This commit is contained in:
Eren Gölge 2021-06-18 14:52:27 +02:00
parent 7dc2177df4
commit 77d57dd301
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/tts/layers/tacotron/tacotron2.py
View File

@ -357,7 +357,7 @@ class Decoder(nn.Module):
if stop_token > self.stop_threshold and t > inputs.shape[0] // 2: if stop_token > self.stop_threshold and t > inputs.shape[0] // 2:
break break
if len(outputs) == self.max_decoder_steps: if len(outputs) == self.max_decoder_steps:
print(" | > Decoder stopped with 'max_decoder_steps") print(f" > Decoder stopped with `max_decoder_steps` {self.max_decoder_steps}")
break break
memory = self._update_memory(decoder_output) memory = self._update_memory(decoder_output)