From 4e44486aa0e87459bed4b0fe423b16e59c76c1a0 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 2 Oct 2022 03:25:28 +0200 Subject: Move stems to transforms --- training/conf/datamodule/iam_extended_paragraphs.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'training/conf/datamodule/iam_extended_paragraphs.yaml') 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 -- cgit v1.2.3-70-g09d2