From 2f8cb4dedfc77827dc855160e9dd999e971cf22c Mon Sep 17 00:00:00 2001 From: Maximilian Ernestus Date: Thu, 18 Jan 2024 14:15:12 +0100 Subject: [PATCH] Add initial_epoch_multiplier to hyperparameter search space of PC --- tuning/hp_search_spaces.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tuning/hp_search_spaces.py b/tuning/hp_search_spaces.py index 304f64eaa..d5913272a 100644 --- a/tuning/hp_search_spaces.py +++ b/tuning/hp_search_spaces.py @@ -89,6 +89,7 @@ def __call__( "discount_factor": trial.suggest_float("gatherer_discount_factor", 0.95, 1.0), "sample": trial.suggest_categorical("gatherer_sample", [True, False]), }, + "initial_epoch_multiplier": trial.suggest_float("initial_epoch_multiplier", 1, 200.0), "initial_comparison_frac": trial.suggest_float("initial_comparison_frac", 0.01, 1.0), "num_iterations": trial.suggest_int("num_iterations", 1, 51), "preference_model_kwargs": {