From b7ca23020cd768addbb3138fabfc62924c578e6c Mon Sep 17 00:00:00 2001 From: Maximilian Ernestus Date: Wed, 28 Feb 2024 13:19:32 +0100 Subject: [PATCH] Fix fragment length for classic control environments. --- tuning/hp_search_spaces.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tuning/hp_search_spaces.py b/tuning/hp_search_spaces.py index 24b1752f4..7af141958 100644 --- a/tuning/hp_search_spaces.py +++ b/tuning/hp_search_spaces.py @@ -135,7 +135,7 @@ def __call__( "active_selection_oversampling": trial.suggest_int("active_selection_oversampling", 1, 11), "comparison_queue_size": trial.suggest_int("comparison_queue_size", 1, 1001), # upper bound determined by total_comparisons=1000 "exploration_frac": trial.suggest_float("exploration_frac", 0.0, 0.5), - "fragment_length": trial.suggest_int("fragment_length", 1, 1001), # trajectories are 1000 steps long + "fragment_length": trial.suggest_int("fragment_length", 1, 201), # trajectories are 1000 steps long "gatherer_kwargs": { "temperature": trial.suggest_float("gatherer_temperature", 0.0, 2.0), "discount_factor": trial.suggest_float("gatherer_discount_factor", 0.95, 1.0),