diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-02-11 22:53:03 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-02-11 22:53:03 +0100 |
| commit | 0c38afdcd787d89bbfb3d9e0698fbf857ce04239 (patch) | |
| tree | 75a53cbcf27fc92a1ba7631d89414eaf4f8afa91 /training | |
| parent | 94ec29dc00b8e19dd7fcdaac0856b6cf00abd469 (diff) | |
feat(config): update default config
Diffstat (limited to 'training')
| -rw-r--r-- | training/conf/config.yaml | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/training/conf/config.yaml b/training/conf/config.yaml index 11bb551..5393c6a 100644 --- a/training/conf/config.yaml +++ b/training/conf/config.yaml @@ -1,15 +1,16 @@ +--- # @package _global_ defaults: - _self_ - callbacks: default - - criterion: label_smoothing + - criterion: cross_entropy - datamodule: iam_extended_paragraphs - hydra: default - logger: wandb - lr_schedulers: - cosine_annealing - - mapping: characters # word_piece + - mapping: characters - model: lit_transformer - network: conv_transformer - optimizers: @@ -31,7 +32,8 @@ logging: INFO # 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 +# 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! |