diff options
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! |