From 441b7484348953deb7c94150675d54583ef5a81a Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 10 Aug 2021 23:25:13 +0200
Subject: Update to config and logging in VQGAN

---
 training/conf/optimizers/madgrad.yaml | 8 ++++++++
 1 file changed, 8 insertions(+)
 create mode 100644 training/conf/optimizers/madgrad.yaml

(limited to 'training/conf/optimizers')

diff --git a/training/conf/optimizers/madgrad.yaml b/training/conf/optimizers/madgrad.yaml
new file mode 100644
index 0000000..a6c059d
--- /dev/null
+++ b/training/conf/optimizers/madgrad.yaml
@@ -0,0 +1,8 @@
+madgrad:
+  _target_: madgrad.MADGRAD
+  lr: 1.0e-3
+  momentum: 0.9
+  weight_decay: 0
+  eps: 1.0e-6
+
+  parameters: network
-- 
cgit v1.2.3-70-g09d2