diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-30 23:15:03 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-30 23:15:03 +0200 |
| commit | 7268035fb9e57342612a8cc50a1fe04e8841ca2f (patch) | |
| tree | 8d4cf3743975bd25f2c04d6a56ff3d4608a7e8d9 /training/conf/callbacks/checkpoint.yaml | |
| parent | 92fc1c7ed2f9f64552be8f71d9b8ab0d5a0a88d4 (diff) | |
attr bug fix, properly loading network
Diffstat (limited to 'training/conf/callbacks/checkpoint.yaml')
| -rw-r--r-- | training/conf/callbacks/checkpoint.yaml | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/training/conf/callbacks/checkpoint.yaml b/training/conf/callbacks/checkpoint.yaml index 9216715..db34cb1 100644 --- a/training/conf/callbacks/checkpoint.yaml +++ b/training/conf/callbacks/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 + 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" - verbose: False - dirpath: "checkpoints/" - filename: "{epoch:02d}" + save_last: true # additionaly always save model from last epoch + mode: min # can be "max" or "min" + verbose: false + dirpath: checkpoints/ + filename: {epoch:02d} |