diff options
Diffstat (limited to 'training/conf/datamodule/iam_extended_paragraphs.yaml')
| -rw-r--r-- | training/conf/datamodule/iam_extended_paragraphs.yaml | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/training/conf/datamodule/iam_extended_paragraphs.yaml b/training/conf/datamodule/iam_extended_paragraphs.yaml index f53e5b6..0921b25 100644 --- a/training/conf/datamodule/iam_extended_paragraphs.yaml +++ b/training/conf/datamodule/iam_extended_paragraphs.yaml @@ -2,7 +2,10 @@ _target_: text_recognizer.data.iam_extended_paragraphs.IAMExtendedParagraphs batch_size: 4 num_workers: 12 train_fraction: 0.8 -pin_memory: false +pin_memory: true transform: transform/paragraphs.yaml test_transform: test_transform/paragraphs.yaml target_transform: target_transform/pad.yaml +mapping: + _target_: text_recognizer.data.mappings.EmnistMapping + extra_symbols: ["\n"] |