diff options
Diffstat (limited to 'text_recognizer/models')
| -rw-r--r-- | text_recognizer/models/transformer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/models/transformer.py b/text_recognizer/models/transformer.py index 7686c4f..f44241d 100644 --- a/text_recognizer/models/transformer.py +++ b/text_recognizer/models/transformer.py @@ -20,7 +20,7 @@ class LitTransformer(LitBase): optimizer_config: DictConfig, lr_scheduler_config: Optional[DictConfig], mapping: EmnistMapping, - max_output_len: int = 451, + max_output_len: int = 682, start_token: str = "<s>", end_token: str = "<e>", pad_token: str = "<p>", |