diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-27 23:11:06 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-27 23:11:06 +0200 |
| commit | 9c7dbb9ca70858b870f74ecf595d3169f0cbc711 (patch) | |
| tree | c342e2c004bb75571a380ef2805049a8fcec3fcc /training/conf/datamodule | |
| parent | 9b8e14d89f0ef2508ed11f994f73af624155fe1d (diff) | |
Rename mapping to tokenizer
Diffstat (limited to 'training/conf/datamodule')
| -rw-r--r-- | training/conf/datamodule/emnist_lines.yaml | 2 | ||||
| -rw-r--r-- | training/conf/datamodule/iam_extended_paragraphs.yaml | 4 | ||||
| -rw-r--r-- | training/conf/datamodule/iam_lines.yaml | 4 |
3 files changed, 5 insertions, 5 deletions
diff --git a/training/conf/datamodule/emnist_lines.yaml b/training/conf/datamodule/emnist_lines.yaml index 218df6c..ce35c3e 100644 --- a/training/conf/datamodule/emnist_lines.yaml +++ b/training/conf/datamodule/emnist_lines.yaml @@ -6,4 +6,4 @@ pin_memory: true transform: transform/lines.yaml test_transform: test_transform/lines.yaml mapping: - _target_: text_recognizer.data.mappings.EmnistMapping + _target_: text_recognizer.data.tokenizer.Tokenizer diff --git a/training/conf/datamodule/iam_extended_paragraphs.yaml b/training/conf/datamodule/iam_extended_paragraphs.yaml index c46714c..64c3964 100644 --- a/training/conf/datamodule/iam_extended_paragraphs.yaml +++ b/training/conf/datamodule/iam_extended_paragraphs.yaml @@ -13,6 +13,6 @@ target_transform: _target_: text_recognizer.data.transforms.pad.Pad max_len: 682 pad_index: 3 -mapping: - _target_: text_recognizer.data.mappings.EmnistMapping +tokenizer: + _target_: text_recognizer.data.tokenizer.Tokenizer extra_symbols: ["\n"] diff --git a/training/conf/datamodule/iam_lines.yaml b/training/conf/datamodule/iam_lines.yaml index 4f1f1b8..f84116d 100644 --- a/training/conf/datamodule/iam_lines.yaml +++ b/training/conf/datamodule/iam_lines.yaml @@ -9,5 +9,5 @@ transform: test_transform: _target_: text_recognizer.data.stems.line.IamLinesStem augment: false -mapping: - _target_: text_recognizer.data.mappings.EmnistMapping +tokenizer: + _target_: text_recognizer.data.tokenizer.Tokenizer |