summaryrefslogtreecommitdiff
path: root/training/conf/callbacks
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-15 00:15:40 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-15 00:15:40 +0200
commit2c9066b685d41ef0ab5ea94e938b8a30b4123656 (patch)
tree341e8d0d73cdde0fe4908421a5e9fd34f16cd8c3 /training/conf/callbacks
parent8bb76745e43c6b4967c8e91ebaf4c4295d0b8d0b (diff)
Update configs
Diffstat (limited to 'training/conf/callbacks')
-rw-r--r--training/conf/callbacks/default.yaml2
-rw-r--r--training/conf/callbacks/lightning/checkpoint.yaml8
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}"