FIX: learning rates

interTwin-eu · Aug 3, 2023 · df19960 · df19960
1 parent 1941428
commit df19960
Showing 1 changed file with 3 additions and 6 deletions.
diff --git 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: [