longer stop_token padding

2026-09-09 11:16:00 +08:00 · 2019-04-01 12:07:24 +02:00
parent e1cd253d65
commit 5212a11836
1 changed files with 1 additions and 1 deletions
@@ -409,7 +409,7 @@ class Decoder(nn.Module):
            stop_flags[2] = t > inputs.shape[1] * 2
            if all(stop_flags):
                stop_count += 1
-                if stop_count > 10:
+                if stop_count > 20:
                    break
            elif len(outputs) == self.max_decoder_steps:
                print("   | > Decoder stopped with 'max_decoder_steps")