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/lr_scheduler/cosine_annealing.yaml | 7 +++++++
 1 file changed, 7 insertions(+)
 create mode 100644 training/conf/lr_scheduler/cosine_annealing.yaml

(limited to 'training/conf/lr_scheduler/cosine_annealing.yaml')

diff --git a/training/conf/lr_scheduler/cosine_annealing.yaml b/training/conf/lr_scheduler/cosine_annealing.yaml
new file mode 100644
index 0000000..62667bb
--- /dev/null
+++ b/training/conf/lr_scheduler/cosine_annealing.yaml
@@ -0,0 +1,7 @@
+_target_: torch.optim.lr_scheduler.CosineAnnealingLR
+T_max: 64
+eta_min: 0.0
+last_epoch: -1
+
+interval: epoch
+monitor: val/loss
-- 
cgit v1.2.3-70-g09d2