From bf6bc0023b1faeda954263e4e41fa92522eb7793 Mon Sep 17 00:00:00 2001 From: EdenWuyifan Date: Wed, 1 May 2024 01:14:14 -0400 Subject: [PATCH] num_rollout_workers to 1 --- alpha_automl/pipeline_search/agent_lab.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/alpha_automl/pipeline_search/agent_lab.py b/alpha_automl/pipeline_search/agent_lab.py index a1614f5b..f3f5f9ce 100644 --- a/alpha_automl/pipeline_search/agent_lab.py +++ b/alpha_automl/pipeline_search/agent_lab.py @@ -22,7 +22,7 @@ def pipeline_search_rllib(game, time_bound, checkpoint_load_folder, checkpoint_s num_cpus = int(ray.available_resources()["CPU"]) # load checkpoint or create a new one - algo = load_rllib_checkpoint(game, checkpoint_load_folder, num_rollout_workers=7) + algo = load_rllib_checkpoint(game, checkpoint_load_folder, num_rollout_workers=1) logger.debug("Create Algo object done") # train model