diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-27 18:16:26 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-27 18:16:26 +0200 |
| commit | ae441a0a390f694cbdda4933d5abfb93f9823b9e (patch) | |
| tree | e3219d61b63465d79295db0c8c42e064f145c033 /text_recognizer | |
| parent | 31da07d71f40eec7e4ef1c053f913b86137922bc (diff) | |
Update default output length
Diffstat (limited to 'text_recognizer')
| -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>", |