From df19960eacfab1e9cf3176a27ea5b99af9668069 Mon Sep 17 00:00:00 2001
From: User3574 <neonikkus@gmail.com>
Date: Thu, 3 Aug 2023 18:34:14 -0400
Subject: [PATCH] FIX: learning rates

---
 use-cases/zebra2horse/pipeline.yaml | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/use-cases/zebra2horse/pipeline.yaml b/use-cases/zebra2horse/pipeline.yaml
index 2a25daaf..6fb02702 100644
--- a/use-cases/zebra2horse/pipeline.yaml
+++ b/use-cases/zebra2horse/pipeline.yaml
@@ -37,22 +37,19 @@ trainer:
       gen_F_optimizer: {
         class_name: "Adam",
         config: {
-          learning_rate: 2e-4,
-          beta_1: 0.5
+            learning_rate: 0.001
         }
       }
       disc_X_optimizer: {
         class_name: "Adam",
         config: {
-          learning_rate: 2e-4,
-          beta_1: 0.5
+            learning_rate: 0.001
         }
       }
       disc_Y_optimizer: {
         class_name: "Adam",
         config: {
-          learning_rate: 2e-4,
-          beta_1: 0.5
+            learning_rate: 0.001
         }
       }
     loggers: [