diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-10-02 03:25:28 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-10-02 03:25:28 +0200 |
| commit | 4e44486aa0e87459bed4b0fe423b16e59c76c1a0 (patch) | |
| tree | a0ce9f72d9000ac8e0959f159a66a7f1dbcf8892 /training/conf/datamodule | |
| parent | 1e0378e1ba1cdab3c064473ef951b97515f28947 (diff) | |
Move stems to transforms
Diffstat (limited to 'training/conf/datamodule')
| -rw-r--r-- | training/conf/datamodule/iam_extended_paragraphs.yaml | 4 | ||||
| -rw-r--r-- | training/conf/datamodule/iam_lines.yaml | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/training/conf/datamodule/iam_extended_paragraphs.yaml b/training/conf/datamodule/iam_extended_paragraphs.yaml index 64c3964..e4ef896 100644 --- a/training/conf/datamodule/iam_extended_paragraphs.yaml +++ b/training/conf/datamodule/iam_extended_paragraphs.yaml @@ -4,10 +4,10 @@ num_workers: 12 train_fraction: 0.8 pin_memory: true transform: - _target_: text_recognizer.data.stems.paragraph.ParagraphStem + _target_: text_recognizer.data.transforms.paragraph.ParagraphStem augment: true test_transform: - _target_: text_recognizer.data.stems.paragraph.ParagraphStem + _target_: text_recognizer.data.transforms.paragraph.ParagraphStem augment: false target_transform: _target_: text_recognizer.data.transforms.pad.Pad diff --git a/training/conf/datamodule/iam_lines.yaml b/training/conf/datamodule/iam_lines.yaml index f84116d..1205c75 100644 --- a/training/conf/datamodule/iam_lines.yaml +++ b/training/conf/datamodule/iam_lines.yaml @@ -4,10 +4,10 @@ num_workers: 12 train_fraction: 0.9 pin_memory: true transform: - _target_: text_recognizer.data.stems.line.IamLinesStem + _target_: text_recognizer.data.transforms.line.IamLinesStem augment: true test_transform: - _target_: text_recognizer.data.stems.line.IamLinesStem + _target_: text_recognizer.data.transforms.line.IamLinesStem augment: false tokenizer: _target_: text_recognizer.data.tokenizer.Tokenizer |