mirror of
https://github.com/wassname/TTS.git
synced 2026-10-28 12:41:20 +08:00
Print max_decoder_steps when model reaches the limit
This commit is contained in:
1 parent
e30f245e06
commit
a5d5bc9063
1 file changed
+1
-1
@@ -357,7 +357,7 @@ class Decoder(nn.Module):
|
||||
if stop_token > self.stop_threshold and t > inputs.shape[0] // 2:
|
||||
break
|
||||
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
|
||||
|
||||
memory = self._update_memory(decoder_output)
|
||||
|
||||
Reference in new issue
Block a user