diff --git a/tuning/hp_search_spaces.py b/tuning/hp_search_spaces.py index d5913272a..60f0cd0ab 100644 --- a/tuning/hp_search_spaces.py +++ b/tuning/hp_search_spaces.py @@ -76,7 +76,7 @@ def __call__( suggest_named_configs=lambda _: ["reward.reward_ensemble"], suggest_config_updates=lambda trial: { "seed": trial.number, - "environment": {"num_vec": 1}, + "environment": {"num_vec": 8}, "total_timesteps": 2e7, "total_comparisons": 1000, "active_selection": True,