diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-03 19:14:16 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-08-03 19:14:16 +0200 |
| commit | 65d5f6c694e73792e40ed693a1381a792da8d277 (patch) | |
| tree | c97067914c18b47108188c51294abd40bb1d6481 /training/conf | |
| parent | bd4bd443f339e95007bfdabf3e060db720f4d4b9 (diff) | |
Fix bugs in converting text in mappings, add missing word_piece arg in datamodule
Diffstat (limited to 'training/conf')
| -rw-r--r-- | training/conf/criterion/label_smoothing.yaml | 2 | ||||
| -rw-r--r-- | training/conf/datamodule/iam_extended_paragraphs.yaml | 1 | ||||
| -rw-r--r-- | training/conf/mapping/emnist.yaml | 2 |
3 files changed, 4 insertions, 1 deletions
diff --git a/training/conf/criterion/label_smoothing.yaml b/training/conf/criterion/label_smoothing.yaml index 684b5bb..45190e5 100644 --- a/training/conf/criterion/label_smoothing.yaml +++ b/training/conf/criterion/label_smoothing.yaml @@ -1,3 +1,3 @@ _target_: text_recognizer.criterions.label_smoothing.LabelSmoothingLoss smoothing: 0.1 -ignore_index: 1002 +ignore_index: 1000 diff --git a/training/conf/datamodule/iam_extended_paragraphs.yaml b/training/conf/datamodule/iam_extended_paragraphs.yaml index 2d1a03e..a2dd293 100644 --- a/training/conf/datamodule/iam_extended_paragraphs.yaml +++ b/training/conf/datamodule/iam_extended_paragraphs.yaml @@ -4,3 +4,4 @@ num_workers: 12 train_fraction: 0.8 augment: true pin_memory: false +word_pieces: true diff --git a/training/conf/mapping/emnist.yaml b/training/conf/mapping/emnist.yaml new file mode 100644 index 0000000..14e966b --- /dev/null +++ b/training/conf/mapping/emnist.yaml @@ -0,0 +1,2 @@ +_target_: text_recognizer.data.emnist_mapping.EmnistMapping +extra_symbols: [ "\n" ] |