summaryrefslogtreecommitdiff
path: root/training/conf/model
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/model')
-rw-r--r--training/conf/model/lit_transformer.yaml5
-rw-r--r--training/conf/model/mapping/word_piece.yaml9
2 files changed, 1 insertions, 13 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>
diff --git a/training/conf/model/mapping/word_piece.yaml b/training/conf/model/mapping/word_piece.yaml
deleted file mode 100644
index 39e2ba4..0000000
--- a/training/conf/model/mapping/word_piece.yaml
+++ /dev/null
@@ -1,9 +0,0 @@
-_target_: text_recognizer.data.mappings.WordPieceMapping
-num_features: 1000
-tokens: iamdb_1kwp_tokens_1000.txt
-lexicon: iamdb_1kwp_lex_1000.txt
-data_dir: null
-use_words: false
-prepend_wordsep: false
-special_tokens: ["<s>", "<e>", "<p>"]
-extra_symbols: ["\n"]