Updated GW150914_D_heterodyne.py

kazewong · Aug 19, 2024 · 645dea1 · 645dea1
1 parent 814c204
commit 645dea1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/example/GW150914_D_heterodyne.py b/example/GW150914_D_heterodyne.py
@@ -121,8 +121,8 @@
     learning_rate=learning_rate,
     n_max_examples=30000,
     n_flow_samples=100000,
-    momentum=30000,
-    batch_size=100,
+    momentum=0.9,
+    batch_size=30000,
     use_global=True,
     train_thinning=1,
     output_thinning=10,