summaryrefslogtreecommitdiff
path: root/text_recognizer/models/transformer.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-27 18:16:26 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-27 18:16:26 +0200
commitae441a0a390f694cbdda4933d5abfb93f9823b9e (patch)
treee3219d61b63465d79295db0c8c42e064f145c033 /text_recognizer/models/transformer.py
parent31da07d71f40eec7e4ef1c053f913b86137922bc (diff)
Update default output length
Diffstat (limited to 'text_recognizer/models/transformer.py')
-rw-r--r--text_recognizer/models/transformer.py2
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>",