Print max_decoder_steps when model reaches the limit

2026-09-09 11:16:00 +08:00 · 2021-06-22 16:21:11 +02:00
parent 7dc2177df4
commit 77d57dd301
1 changed files with 1 additions and 1 deletions
@@ -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)