diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-10 18:06:54 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-10 18:06:54 +0200 |
| commit | b69254ce3135c112e29f7f1c986b7f0817da0c33 (patch) | |
| tree | e596a719b445c0cbbd4108079206ec9d14de1437 /training/conf/datamodule | |
| parent | cba94bdcab90f288dd1172607500ba2b28279736 (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, 4 insertions, 5 deletions
diff --git a/training/conf/datamodule/iam_extended_paragraphs.yaml b/training/conf/datamodule/iam_extended_paragraphs.yaml index a0ffe56..6fa8206 100644 --- a/training/conf/datamodule/iam_extended_paragraphs.yaml +++ b/training/conf/datamodule/iam_extended_paragraphs.yaml @@ -2,7 +2,6 @@ _target_: text_recognizer.data.iam_extended_paragraphs.IAMExtendedParagraphs batch_size: 4 num_workers: 12 train_fraction: 0.8 -augment: true pin_memory: false -word_pieces: true -resize: null +transform: transform/paragraphs.yaml +test_transform: transform/paragraphs.yaml diff --git a/training/conf/datamodule/iam_lines.yaml b/training/conf/datamodule/iam_lines.yaml index d3c2af8..36e7093 100644 --- a/training/conf/datamodule/iam_lines.yaml +++ b/training/conf/datamodule/iam_lines.yaml @@ -2,6 +2,6 @@ _target_: text_recognizer.data.iam_lines.IAMLines batch_size: 8 num_workers: 12 train_fraction: 0.8 -augment: true pin_memory: false -# word_pieces: true +transform: transform/iam_lines.yaml +test_transform: test_transform/iam_lines.yaml |