summaryrefslogtreecommitdiff
path: root/training/conf/datamodule
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/datamodule')
-rw-r--r--training/conf/datamodule/iam_extended_paragraphs.yaml4
-rw-r--r--training/conf/datamodule/iam_lines.yaml4
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