Add initial_epoch_multiplier to hyperparameter search space of PC

HumanCompatibleAI · Jan 18, 2024 · 2f8cb4d · 2f8cb4d
1 parent 8e31713
commit 2f8cb4d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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": {