diff --git a/tests/test_flow_model.py b/tests/test_flow_model.py index f0f13651..875cebdd 100644 --- a/tests/test_flow_model.py +++ b/tests/test_flow_model.py @@ -154,7 +154,7 @@ def test_RealNVP_gaussian(): RealNVP = model_nf.RealNVPModel(ndim, standardize=True) RealNVP.fit(samples, epochs=epochs, verbose=True) - nsamples = 10000 + nsamples = 20000 RealNVP.temperature = 1.0 flow_samples = RealNVP.sample(nsamples) sample_var = jnp.var(flow_samples, axis=0)