From 969e1d5e179d9c42ffae0c9b12c9bd3be6091360 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 4 Jul 2021 22:58:07 +0200
Subject: Add wandb callbacks

---
 training/conf/callbacks/checkpoint.yaml | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

(limited to 'training/conf/callbacks/checkpoint.yaml')

diff --git a/training/conf/callbacks/checkpoint.yaml b/training/conf/callbacks/checkpoint.yaml
index f3beb1b..9216715 100644
--- a/training/conf/callbacks/checkpoint.yaml
+++ b/training/conf/callbacks/checkpoint.yaml
@@ -1,6 +1,9 @@
-checkpoint:
-  type: ModelCheckpoint
-  args:
-    monitor: val_loss
-    mode: min
-    save_last: true
+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"
+  verbose: False
+  dirpath: "checkpoints/"
+  filename: "{epoch:02d}"
-- 
cgit v1.2.3-70-g09d2