summaryrefslogtreecommitdiff
path: root/training/conf/model/lit_transformer.yaml
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-03 00:33:33 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-03 00:33:33 +0200
commitbecc8e1380a36f45a8dadf5a7cc6c7b922fe8dff (patch)
treefc07d6b8304d85d6c9c0e5e1638f0f4a2bc07e46 /training/conf/model/lit_transformer.yaml
parente6a5ef0330fb35f5c1c79872d57c3293127d4aa7 (diff)
Add experiment configs
Diffstat (limited to 'training/conf/model/lit_transformer.yaml')
-rw-r--r--training/conf/model/lit_transformer.yaml2
1 files changed, 1 insertions, 1 deletions
diff --git a/training/conf/model/lit_transformer.yaml b/training/conf/model/lit_transformer.yaml
index 0ec3b8a..5172533 100644
--- a/training/conf/model/lit_transformer.yaml
+++ b/training/conf/model/lit_transformer.yaml
@@ -1,5 +1,5 @@
_target_: text_recognizer.models.transformer.TransformerLitModel
-max_output_len: 451
+max_output_len: 682
start_token: <s>
end_token: <e>
pad_token: <p>