Change test parameters.

astro-informatics · Oct 13, 2023 · 00be79c · 00be79c
1 parent 62e572a
commit 00be79c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_flow_model.py b/tests/test_flow_model.py
@@ -101,7 +101,7 @@ def test_RealNVP_gaussian():
     # Define the number of dimensions and the mean of the Gaussian
     ndim = 2
     num_samples = 10000
-    epochs = 50
+    epochs = 80
     # Initialize a PRNG key (you can use any valid key)
     key = jax.random.PRNGKey(0)
     mean = jnp.zeros(ndim)