diff options
Diffstat (limited to 'training/conf/datamodule/iam_lines.yaml')
| -rw-r--r-- | training/conf/datamodule/iam_lines.yaml | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/training/conf/datamodule/iam_lines.yaml b/training/conf/datamodule/iam_lines.yaml index 85a023b..4f1f1b8 100644 --- a/training/conf/datamodule/iam_lines.yaml +++ b/training/conf/datamodule/iam_lines.yaml @@ -3,7 +3,11 @@ batch_size: 8 num_workers: 12 train_fraction: 0.9 pin_memory: true -transform: transform/lines.yaml -test_transform: test_transform/lines.yaml +transform: + _target_: text_recognizer.data.stems.line.IamLinesStem + augment: true +test_transform: + _target_: text_recognizer.data.stems.line.IamLinesStem + augment: false mapping: _target_: text_recognizer.data.mappings.EmnistMapping |