training/conf/config.yaml


1
2
3
4
5
6
7
8
9
10
11
12
13
14

defaults:
  - network: vqvae
  - criterion: mse
  - optimizer: madgrad
  - lr_scheduler: one_cycle
  - model: lit_vqvae
  - dataset: iam_extended_paragraphs
  - trainer: default
  - callbacks:
    - checkpoint
    - learning_rate_monitor

load_checkpoint: null 
logging: INFO