summaryrefslogtreecommitdiff
path: root/training/conf/model/lit_transformer.yaml
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-29 23:59:52 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-29 23:59:52 +0200
commit34098ccbbbf6379c0bd29a987440b8479c743746 (patch)
treea8c68e3036503049fc7034c677ec855465f7a8e0 /training/conf/model/lit_transformer.yaml
parentc032ffb05a7ed86f8fe5d596f94e8997c558cae8 (diff)
Configs, refactor with attrs, fix attr bug in iam
Diffstat (limited to 'training/conf/model/lit_transformer.yaml')
-rw-r--r--training/conf/model/lit_transformer.yaml4
1 files changed, 4 insertions, 0 deletions
diff --git a/training/conf/model/lit_transformer.yaml b/training/conf/model/lit_transformer.yaml
new file mode 100644
index 0000000..4e04b85
--- /dev/null
+++ b/training/conf/model/lit_transformer.yaml
@@ -0,0 +1,4 @@
+_target_: text_recognizer.models.transformer.TransformerLitModel
+interval: null
+monitor: val/loss
+ignore_tokens: ["<s>", "<e>", "<p>"]