From 263f2b7158d76bc0adad45309625910c0fa7b1fe Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Fri, 6 Aug 2021 14:19:37 +0200
Subject: Remove lr args from model, add Cosine lr, fix to vqvae stack

---
 training/conf/experiment/vqvae.yaml | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

(limited to 'training/conf/experiment')

diff --git a/training/conf/experiment/vqvae.yaml b/training/conf/experiment/vqvae.yaml
index eb40f3b..7a9e643 100644
--- a/training/conf/experiment/vqvae.yaml
+++ b/training/conf/experiment/vqvae.yaml
@@ -5,6 +5,7 @@ defaults:
   - override /criterion: mse
   - override /model: lit_vqvae
   - override /callbacks: wandb_vae
+  - override /lr_scheduler: cosine_annealing
 
 trainer:
   max_epochs: 64
@@ -13,11 +14,11 @@ trainer:
 datamodule:
   batch_size: 16
 
-lr_scheduler:
-  epochs: 64
-  steps_per_epoch: 1245
+# lr_scheduler:
+  # epochs: 64
+  # steps_per_epoch: 1245
 
-optimizer:
-  lr: 1.0e-3
+# optimizer:
+  # lr: 1.0e-3
 
 summary: [1, 576, 640]
-- 
cgit v1.2.3-70-g09d2