summaryrefslogtreecommitdiff
path: root/training/conf/config.yaml
blob: e783efd17426198795c8a82bc2a4c5205ac2acfc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# @package _global_

defaults:
  - _self_
  - callbacks: default
  - criterion: label_smoothing
  - datamodule: iam_extended_paragraphs
  - hydra: default
  - logger: wandb
  - lr_schedulers: 
      - cosine_annealing
  - mapping: characters # word_piece
  - model: lit_transformer
  - network: conv_transformer
  - optimizers: 
      - madgrad
  - trainer: default

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

# datamodule:
#   word_pieces: false

# model:
#   max_output_len: 682

# path to original working directory
# hydra hijacks working directory by changing it to the current log directory,
# so it's useful to have this path as a special variable
# learn more here: https://hydra.cc/docs/next/tutorials/basic/running_your_app/working_directory
work_dir: ${hydra:runtime.cwd}

# use `python run.py debug=true` for easy debugging!
# this will run 1 train, val and test loop with only 1 batch
# equivalent to running `python run.py trainer.fast_dev_run=true`
# (this is placed here just for easier access from command line)
debug: False

# pretty print config at the start of the run using Rich library
print_config: false

# disable python warnings if they annoy you
ignore_warnings: true

summary: null  # [1, 576, 640]