From 0ba945e84d11a07ac95fdf8495f2ff278215adb9 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 27 Jun 2021 16:52:55 +0200
Subject: Changed structure of callbacks

---
 training/conf/callbacks/checkpoint.yaml            |  5 +++++
 training/conf/callbacks/default.yaml               | 14 -------------
 training/conf/callbacks/early_stopping.yaml        |  5 +++++
 training/conf/callbacks/learning_rate_monitor.yaml |  3 +++
 training/conf/callbacks/swa.yaml                   | 23 +++++++---------------
 5 files changed, 20 insertions(+), 30 deletions(-)
 create mode 100644 training/conf/callbacks/checkpoint.yaml
 delete mode 100644 training/conf/callbacks/default.yaml
 create mode 100644 training/conf/callbacks/early_stopping.yaml
 create mode 100644 training/conf/callbacks/learning_rate_monitor.yaml

(limited to 'training/conf/callbacks')

diff --git a/training/conf/callbacks/checkpoint.yaml b/training/conf/callbacks/checkpoint.yaml
new file mode 100644
index 0000000..afc536f
--- /dev/null
+++ b/training/conf/callbacks/checkpoint.yaml
@@ -0,0 +1,5 @@
+type: ModelCheckpoint
+args:
+    monitor: val_loss
+    mode: min
+    save_last: true
diff --git a/training/conf/callbacks/default.yaml b/training/conf/callbacks/default.yaml
deleted file mode 100644
index 74dc30c..0000000
--- a/training/conf/callbacks/default.yaml
+++ /dev/null
@@ -1,14 +0,0 @@
-# @package _group_
-- type: ModelCheckpoint
-  args:
-      monitor: val_loss
-      mode: min
-      save_last: true
-- type: LearningRateMonitor
-  args:
-      logging_interval: step
-# - type: EarlyStopping
-#   args:
-#       monitor: val_loss
-#       mode: min
-#       patience: 10
diff --git a/training/conf/callbacks/early_stopping.yaml b/training/conf/callbacks/early_stopping.yaml
new file mode 100644
index 0000000..caab824
--- /dev/null
+++ b/training/conf/callbacks/early_stopping.yaml
@@ -0,0 +1,5 @@
+type: EarlyStopping
+args:
+    monitor: val_loss
+    mode: min
+    patience: 10
diff --git a/training/conf/callbacks/learning_rate_monitor.yaml b/training/conf/callbacks/learning_rate_monitor.yaml
new file mode 100644
index 0000000..003ab7a
--- /dev/null
+++ b/training/conf/callbacks/learning_rate_monitor.yaml
@@ -0,0 +1,3 @@
+type: LearningRateMonitor
+args:
+    logging_interval: step
diff --git a/training/conf/callbacks/swa.yaml b/training/conf/callbacks/swa.yaml
index 144ad6e..279ca69 100644
--- a/training/conf/callbacks/swa.yaml
+++ b/training/conf/callbacks/swa.yaml
@@ -1,16 +1,7 @@
-# @package _group_
-- type: ModelCheckpoint
-  args:
-      monitor: val_loss
-      mode: min
-      save_last: true
-- type: StochasticWeightAveraging
-  args:
-      swa_epoch_start: 0.8
-      swa_lrs: 0.05
-      annealing_epochs: 10
-      annealing_strategy: cos
-      device: null
-- type: LearningRateMonitor
-  args:
-      logging_interval: step
+type: StochasticWeightAveraging
+args:
+    swa_epoch_start: 0.8
+    swa_lrs: 0.05
+    annealing_epochs: 10
+    annealing_strategy: cos
+    device: null
-- 
cgit v1.2.3-70-g09d2