From fbfbe5eaff7b61354ad3b57921b57c4d969cc936 Mon Sep 17 00:00:00 2001 From: ThibeauWouters Date: Mon, 27 Nov 2023 11:28:17 -0800 Subject: [PATCH] added popsize kwarg to EvolutionaryOptimizer --- src/jimgw/likelihood.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/jimgw/likelihood.py b/src/jimgw/likelihood.py index 579eca5b..3c7f93c0 100644 --- a/src/jimgw/likelihood.py +++ b/src/jimgw/likelihood.py @@ -378,7 +378,7 @@ def maximize_likelihood( y = jax.jit(jax.vmap(y)) print("Starting the optimizer") - optimizer = EvolutionaryOptimizer(len(bounds), verbose=True) + optimizer = EvolutionaryOptimizer(len(bounds), popsize=set_nwalkers, verbose=True) state = optimizer.optimize(y, bounds, n_loops=n_loops) best_fit = optimizer.get_result()[0] return prior.add_name(best_fit, transform_name=True, transform_value=True)