fix: hypersweeper iteration counter and float fidelity for NePS

becktepe · Oct 8, 2024 · 6e32507 · 6e32507
1 parent c0fe63f
commit 6e32507
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/hydra_plugins/hyper_neps/hyper_neps.py b/hydra_plugins/hyper_neps/hyper_neps.py
@@ -92,7 +92,7 @@ def make_neps(configspace, hyper_neps_args):
 
     dict_search_space = get_dict_from_configspace(configspace)
 
-    dict_search_space[hyper_neps_args["fidelity_variable"]] = neps.IntegerParameter(
+    dict_search_space[hyper_neps_args["fidelity_variable"]] = neps.FloatParameter(
         lower=hyper_neps_args["min_budget"],
         upper=hyper_neps_args["max_budget"],
         is_fidelity=True

diff --git a/hydra_plugins/hypersweeper/hypersweeper_sweeper.py b/hydra_plugins/hypersweeper/hypersweeper_sweeper.py
@@ -485,6 +485,7 @@ def run(self, verbose=False):
 
             self.opt_time += time.time() - opt_time_start
             done = trial_termination or budget_termination
+            self.iteration += 1
 
         total_time = time.time() - self.start
         inc_config, inc_performance = self.get_incumbent()