From 8177b5210fbbe11bd361dedbd5b4a4e1950bdb2e Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Fri, 10 Jun 2022 00:36:51 +0200
Subject: Update configs

---
 training/conf/model/lit_transformer.yaml | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'training/conf/model')

diff --git a/training/conf/model/lit_transformer.yaml b/training/conf/model/lit_transformer.yaml
index 5172533..c1491ec 100644
--- a/training/conf/model/lit_transformer.yaml
+++ b/training/conf/model/lit_transformer.yaml
@@ -1,5 +1,8 @@
-_target_: text_recognizer.models.transformer.TransformerLitModel
+_target_: text_recognizer.models.LitTransformer
 max_output_len: 682
 start_token: <s>
 end_token: <e>
 pad_token: <p>
+mapping:
+  _target_: text_recognizer.data.mappings.EmnistMapping
+  extra_symbols: ["\n"]
-- 
cgit v1.2.3-70-g09d2