summaryrefslogtreecommitdiff
path: root/training/conf/config.yaml
blob: 93215ed8857b7cfb4d6af14bb3490fa26c23d485 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
defaults:
  - callbacks: wandb_ocr
  - criterion: label_smoothing
  - dataset: iam_extended_paragraphs
  - hydra: default
  - lr_scheduler: one_cycle
  - mapping: word_piece
  - model: lit_transformer
  - network: conv_transformer
  - optimizer: madgrad
  - trainer: default

seed: 4711
tune: false
train: true
test: true
logging: INFO