diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-15 00:15:40 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-15 00:15:40 +0200 |
commit | 2c9066b685d41ef0ab5ea94e938b8a30b4123656 (patch) | |
tree | 341e8d0d73cdde0fe4908421a5e9fd34f16cd8c3 /training/conf/callbacks | |
parent | 8bb76745e43c6b4967c8e91ebaf4c4295d0b8d0b (diff) |
Update configs
Diffstat (limited to 'training/conf/callbacks')
-rw-r--r-- | training/conf/callbacks/default.yaml | 2 | ||||
-rw-r--r-- | training/conf/callbacks/lightning/checkpoint.yaml | 8 |
2 files changed, 5 insertions, 5 deletions
diff --git a/training/conf/callbacks/default.yaml b/training/conf/callbacks/default.yaml index 57c10a6..4d8e399 100644 --- a/training/conf/callbacks/default.yaml +++ b/training/conf/callbacks/default.yaml @@ -2,5 +2,5 @@ defaults: - lightning/checkpoint - lightning/learning_rate_monitor - wandb/watch - - wandb/config + - wandb/config - wandb/checkpoints diff --git a/training/conf/callbacks/lightning/checkpoint.yaml b/training/conf/callbacks/lightning/checkpoint.yaml index b4101d8..9acd64f 100644 --- a/training/conf/callbacks/lightning/checkpoint.yaml +++ b/training/conf/callbacks/lightning/checkpoint.yaml @@ -1,9 +1,9 @@ model_checkpoint: _target_: pytorch_lightning.callbacks.ModelCheckpoint - monitor: val/loss # name of the logged metric which determines when model is improving - save_top_k: 1 # save k best models (determined by above metric) - save_last: true # additionaly always save model from last epoch - mode: min # can be "max" or "min" + monitor: val/cer + save_top_k: 1 + save_last: true + mode: min verbose: false dirpath: checkpoints/ filename: "{epoch:02d}" |