Print `max_decoder_steps` when model reaches the limit

2021-06-18 14:52:27 +02:00 · 2021-06-18 14:52:27 +02:00 · 77d57dd301
parent 7dc2177df4
commit 77d57dd301
1 changed files with 1 additions and 1 deletions
--- a/TTS/tts/layers/tacotron/tacotron2.py
+++ b/TTS/tts/layers/tacotron/tacotron2.py
@ -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)