From a65d3ec18a5541cec5297769f1027422975a62bc Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 3 Sep 2023 01:13:37 +0200
Subject: Update confs and callbacks

---
 training/conf/callbacks/default.yaml           | 1 -
 training/conf/callbacks/lightning/swa.yaml     | 7 +++++++
 training/conf/callbacks/wandb/captions.yaml    | 2 +-
 training/conf/callbacks/wandb/checkpoints.yaml | 4 ----
 training/conf/callbacks/wandb/swa.yaml         | 7 -------
 5 files changed, 8 insertions(+), 13 deletions(-)
 create mode 100644 training/conf/callbacks/lightning/swa.yaml
 delete mode 100644 training/conf/callbacks/wandb/checkpoints.yaml
 delete mode 100644 training/conf/callbacks/wandb/swa.yaml

(limited to 'training/conf/callbacks')

diff --git a/training/conf/callbacks/default.yaml b/training/conf/callbacks/default.yaml
index 4d8e399..9b1347b 100644
--- a/training/conf/callbacks/default.yaml
+++ b/training/conf/callbacks/default.yaml
@@ -3,4 +3,3 @@ defaults:
   - lightning/learning_rate_monitor
   - wandb/watch
   - wandb/config
-  - wandb/checkpoints
diff --git a/training/conf/callbacks/lightning/swa.yaml b/training/conf/callbacks/lightning/swa.yaml
new file mode 100644
index 0000000..73f8c66
--- /dev/null
+++ b/training/conf/callbacks/lightning/swa.yaml
@@ -0,0 +1,7 @@
+stochastic_weight_averaging:
+  _target_: pytorch_lightning.callbacks.StochasticWeightAveraging
+  swa_epoch_start: 0.8
+  swa_lrs: 0.05
+  annealing_epochs: 10
+  annealing_strategy: cos
+  device: null
diff --git a/training/conf/callbacks/wandb/captions.yaml b/training/conf/callbacks/wandb/captions.yaml
index 3215a90..64f44c2 100644
--- a/training/conf/callbacks/wandb/captions.yaml
+++ b/training/conf/callbacks/wandb/captions.yaml
@@ -1,4 +1,4 @@
 log_text_predictions:
-  _target_: callbacks.wandb.ImageToCaptionLogger
+  _target_: callbacks.wandb.ImageToCaption
   num_samples: 8
   on_train: true
diff --git a/training/conf/callbacks/wandb/checkpoints.yaml b/training/conf/callbacks/wandb/checkpoints.yaml
deleted file mode 100644
index b9a3fd7..0000000
--- a/training/conf/callbacks/wandb/checkpoints.yaml
+++ /dev/null
@@ -1,4 +0,0 @@
-upload_ckpts_as_artifact:
-  _target_: callbacks.wandb.UploadCheckpointsAsArtifact
-  ckpt_dir: checkpoints/
-  upload_best_only: true
diff --git a/training/conf/callbacks/wandb/swa.yaml b/training/conf/callbacks/wandb/swa.yaml
deleted file mode 100644
index 73f8c66..0000000
--- a/training/conf/callbacks/wandb/swa.yaml
+++ /dev/null
@@ -1,7 +0,0 @@
-stochastic_weight_averaging:
-  _target_: pytorch_lightning.callbacks.StochasticWeightAveraging
-  swa_epoch_start: 0.8
-  swa_lrs: 0.05
-  annealing_epochs: 10
-  annealing_strategy: cos
-  device: null
-- 
cgit v1.2.3-70-g09d2