summaryrefslogtreecommitdiff
path: root/training/conf/datamodule/iam_extended_paragraphs.yaml
blob: 64c396462c244dfa790d8678189bf5f219d4933c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
_target_: text_recognizer.data.iam_extended_paragraphs.IAMExtendedParagraphs
batch_size: 4
num_workers: 12
train_fraction: 0.8
pin_memory: true
transform:
  _target_: text_recognizer.data.stems.paragraph.ParagraphStem
  augment: true
test_transform:
  _target_: text_recognizer.data.stems.paragraph.ParagraphStem
  augment: false
target_transform:
  _target_: text_recognizer.data.transforms.pad.Pad
  max_len: 682
  pad_index: 3
tokenizer:
  _target_: text_recognizer.data.tokenizer.Tokenizer
  extra_symbols: ["\n"]