diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-01-26 23:22:13 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-01-26 23:22:13 +0100 |
| commit | 69e93847bb0adf6e819f48e70e6a88da4cf0b7f6 (patch) | |
| tree | b7e533674d51f8e34a9318b7a4f7344aae56f5de /training/conf/config.yaml | |
| parent | 68640a70a769568fd571e50322d6e3fa40c78271 (diff) | |
fix: nuke madgrad, add radam
Diffstat (limited to 'training/conf/config.yaml')
| -rw-r--r-- | training/conf/config.yaml | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/training/conf/config.yaml b/training/conf/config.yaml index e783efd..11bb551 100644 --- a/training/conf/config.yaml +++ b/training/conf/config.yaml @@ -7,13 +7,13 @@ defaults: - datamodule: iam_extended_paragraphs - hydra: default - logger: wandb - - lr_schedulers: + - lr_schedulers: - cosine_annealing - mapping: characters # word_piece - model: lit_transformer - network: conv_transformer - - optimizers: - - madgrad + - optimizers: + - radam - trainer: default seed: 4711 |