From 2c9066b685d41ef0ab5ea94e938b8a30b4123656 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 15 Jun 2022 00:15:40 +0200 Subject: Update configs --- training/conf/callbacks/lightning/checkpoint.yaml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'training/conf/callbacks/lightning/checkpoint.yaml') 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}" -- cgit v1.2.3-70-g09d2