diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-27 01:44:49 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-09-27 01:44:49 +0200 |
commit | 9b8e14d89f0ef2508ed11f994f73af624155fe1d (patch) | |
tree | 10d9c138f4449646c8b5c0f95003ba86b831d04d /training/conf/datamodule/iam_extended_paragraphs.yaml | |
parent | 63376b1c2da81c23ad5239f908b640cd42a514c7 (diff) |
Update data modules
Diffstat (limited to 'training/conf/datamodule/iam_extended_paragraphs.yaml')
-rw-r--r-- | training/conf/datamodule/iam_extended_paragraphs.yaml | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/training/conf/datamodule/iam_extended_paragraphs.yaml b/training/conf/datamodule/iam_extended_paragraphs.yaml index 0921b25..c46714c 100644 --- a/training/conf/datamodule/iam_extended_paragraphs.yaml +++ b/training/conf/datamodule/iam_extended_paragraphs.yaml @@ -3,9 +3,16 @@ batch_size: 4 num_workers: 12 train_fraction: 0.8 pin_memory: true -transform: transform/paragraphs.yaml -test_transform: test_transform/paragraphs.yaml -target_transform: target_transform/pad.yaml +transform: + _target_: text_recognizer.data.stems.paragraph.ParagraphStem + augment: true +test_transform: + _target_: text_recognizer.data.stems.paragraph.ParagraphStem + augment: false +target_transform: + _target_: text_recognizer.data.transforms.pad.Pad + max_len: 682 + pad_index: 3 mapping: _target_: text_recognizer.data.mappings.EmnistMapping extra_symbols: ["\n"] |