From 0e79082555acadc23ee9fb18e1ba1674a1003a6b Mon Sep 17 00:00:00 2001 From: alicjapolanska Date: Tue, 24 Oct 2023 11:33:59 +0100 Subject: [PATCH] Increase sample number for flows test. --- tests/test_flow_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)