summaryrefslogtreecommitdiff
path: root/training
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-11 22:10:34 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-10-11 22:10:34 +0200
commite0c2c84aba3cdbdde832e90a4a294d24032505a6 (patch)
tree5479415e16b27c2f0c2c9115d93484a154de15ba /training
parent191b74d3d9ef2a46933e79a1688bb5c99a7bae93 (diff)
Refactor callback configs
Diffstat (limited to 'training')
-rw-r--r--training/conf/callbacks/default.yaml7
-rw-r--r--training/conf/callbacks/htr.yaml5
-rw-r--r--training/conf/callbacks/vae.yaml5
3 files changed, 7 insertions, 10 deletions
diff --git a/training/conf/callbacks/default.yaml b/training/conf/callbacks/default.yaml
index c184039..57c10a6 100644
--- a/training/conf/callbacks/default.yaml
+++ b/training/conf/callbacks/default.yaml
@@ -1,3 +1,6 @@
defaults:
- - lightning: checkpoint
- - lightning: learning_rate_monitor
+ - lightning/checkpoint
+ - lightning/learning_rate_monitor
+ - wandb/watch
+ - wandb/config
+ - wandb/checkpoints
diff --git a/training/conf/callbacks/htr.yaml b/training/conf/callbacks/htr.yaml
index 51c68c5..ebcb0fa 100644
--- a/training/conf/callbacks/htr.yaml
+++ b/training/conf/callbacks/htr.yaml
@@ -1,6 +1,3 @@
defaults:
- default
- - wandb: watch
- - wandb: config
- - wandb: checkpoints
- - wandb: htr_predictions
+ - wandb/predictions
diff --git a/training/conf/callbacks/vae.yaml b/training/conf/callbacks/vae.yaml
index eec2c1f..52adf69 100644
--- a/training/conf/callbacks/vae.yaml
+++ b/training/conf/callbacks/vae.yaml
@@ -1,6 +1,3 @@
defaults:
- default
- - wandb: watch
- - wandb: checkpoints
- - wandb: reconstructions
- - wandb: config
+ - wandb/reconstructions