diff --git a/example/GW150914_D_heterodyne.py b/example/GW150914_D_heterodyne.py
index cb89b7e5..91e04ba2 100644
--- 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,