summaryrefslogtreecommitdiff
path: root/training/conf/config.yaml
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-27 16:52:55 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-27 16:52:55 +0200
commit0ba945e84d11a07ac95fdf8495f2ff278215adb9 (patch)
treeb22bcecf37416fbdfbe6a5ef178471e1a5751c3e /training/conf/config.yaml
parent22e36513dd43d2e2ca82ca28a1ea757c5663676a (diff)
Changed structure of callbacks
Diffstat (limited to 'training/conf/config.yaml')
-rw-r--r--training/conf/config.yaml12
1 files changed, 7 insertions, 5 deletions
diff --git a/training/conf/config.yaml b/training/conf/config.yaml
index 11adeb7..c413a1a 100644
--- a/training/conf/config.yaml
+++ b/training/conf/config.yaml
@@ -1,6 +1,8 @@
defaults:
- - network: vqvae
- - model: lit_vqvae
- - dataset: iam_extended_paragraphs
- - trainer: default
- - callbacks: default
+ - network: vqvae
+ - model: lit_vqvae
+ - dataset: iam_extended_paragraphs
+ - trainer: default
+ - callbacks:
+ - checkpoint
+ - learning_rate_monitor