Fix for "Problem with HyperBand setup given a total budget #1179"

automl · Jan 13, 2025 · a2e7f98 · a2e7f98
1 parent 06723ed
commit a2e7f98
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/smac/intensifier/hyperband_utils.py b/smac/intensifier/hyperband_utils.py
@@ -44,7 +44,11 @@ def determine_HB(min_budget: float, max_budget: float, eta: int = 3) -> dict:
         _max_iterations[i] = max_iter + 1
 
     total_trials = np.sum([np.sum(v) for v in _n_configs_in_stage.values()])
-    total_budget = np.sum([np.sum(v) for v in _budgets_in_stage.values()])
+
+    total_budget = 0
+    for stage in _n_configs_in_stage.keys():
+        for b, c in zip(_budgets_in_stage[stage], _n_configs_in_stage[stage]):
+            total_budget += b * c
 
     return {
         "max_iterations": _max_iterations,