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/optimizer/madgrad.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'training/conf/optimizer')

diff --git a/training/conf/optimizer/madgrad.yaml b/training/conf/optimizer/madgrad.yaml
index 46b2fff..458b116 100644
--- a/training/conf/optimizer/madgrad.yaml
+++ b/training/conf/optimizer/madgrad.yaml
@@ -1,5 +1,5 @@
 _target_: madgrad.MADGRAD
-lr: 2.0e-4
+lr: 3.0e-4
 momentum: 0.9
 weight_decay: 0
 eps: 1.0e-6
-- 
cgit v1.2.3-70-g09d2