summaryrefslogtreecommitdiff
path: root/training/conf/model/lit_transformer.yaml
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/model/lit_transformer.yaml')
-rw-r--r--training/conf/model/lit_transformer.yaml5
1 files changed, 1 insertions, 4 deletions
diff --git a/training/conf/model/lit_transformer.yaml b/training/conf/model/lit_transformer.yaml
index 5341d8e..6ffde4e 100644
--- a/training/conf/model/lit_transformer.yaml
+++ b/training/conf/model/lit_transformer.yaml
@@ -1,8 +1,5 @@
-defaults:
- - mapping: word_piece
-
_target_: text_recognizer.models.transformer.TransformerLitModel
-interval: null
+interval: step
monitor: val/loss
ignore_tokens: [ <s>, <e>, <p> ]
start_token: <s>