diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-11 23:10:56 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-11 23:10:56 +0200 |
| commit | 8fe4b36bf22281c84c4afee811b3435f3b50686d (patch) | |
| tree | ab378f98c03ee63943a6a026dcbf869d7a39df23 /training/conf/datamodule | |
| parent | 3fa2733b697bfe1c1f8a6031ae7f1f66343d5a5b (diff) | |
Update configs
Diffstat (limited to 'training/conf/datamodule')
| -rw-r--r-- | training/conf/datamodule/iam_extended_paragraphs.yaml | 5 | ||||
| -rw-r--r-- | training/conf/datamodule/iam_lines.yaml | 4 |
2 files changed, 7 insertions, 2 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"] diff --git a/training/conf/datamodule/iam_lines.yaml b/training/conf/datamodule/iam_lines.yaml index 19c72f8..85a023b 100644 --- a/training/conf/datamodule/iam_lines.yaml +++ b/training/conf/datamodule/iam_lines.yaml @@ -1,7 +1,9 @@ _target_: text_recognizer.data.iam_lines.IAMLines batch_size: 8 num_workers: 12 -train_fraction: 0.8 +train_fraction: 0.9 pin_memory: true transform: transform/lines.yaml test_transform: test_transform/lines.yaml +mapping: + _target_: text_recognizer.data.mappings.EmnistMapping |