minor update to CMA-ES

Former-commit-id: 6b9231e
zuoxingdong · May 14, 2019 · 2d2b34f · 2d2b34f
1 parent 12e3d3a
commit 2d2b34f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/baselines/cmaes/experiment.py b/baselines/cmaes/experiment.py
@@ -106,7 +106,7 @@ def run(config, seed, device, logdir):
         for generation in range(config['train.generations']):
             start_time = time.perf_counter()
             solutions = es.ask()
-            out = list(executor.map(fitness, solutions, chunksize=4))
+            out = list(executor.map(fitness, solutions, chunksize=2))
             Rs, Hs = zip(*out)
             es.tell(solutions, [-R for R in Rs])
             logger = Logger()