summaryrefslogtreecommitdiff
path: root/training/conf/callbacks
diff options
context:
space:
mode:
Diffstat (limited to 'training/conf/callbacks')
-rw-r--r--training/conf/callbacks/checkpoint.yaml5
-rw-r--r--training/conf/callbacks/default.yaml14
-rw-r--r--training/conf/callbacks/early_stopping.yaml5
-rw-r--r--training/conf/callbacks/learning_rate_monitor.yaml3
-rw-r--r--training/conf/callbacks/swa.yaml23
5 files changed, 20 insertions, 30 deletions
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