diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-11 22:10:34 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-11 22:10:34 +0200 |
commit | e0c2c84aba3cdbdde832e90a4a294d24032505a6 (patch) | |
tree | 5479415e16b27c2f0c2c9115d93484a154de15ba /training/conf/callbacks | |
parent | 191b74d3d9ef2a46933e79a1688bb5c99a7bae93 (diff) |
Refactor callback configs
Diffstat (limited to 'training/conf/callbacks')
-rw-r--r-- | training/conf/callbacks/default.yaml | 7 | ||||
-rw-r--r-- | training/conf/callbacks/htr.yaml | 5 | ||||
-rw-r--r-- | training/conf/callbacks/vae.yaml | 5 |
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 |