Resource value should be based on bracket iteration
@@ -112,11 +112,13 @@ class HyperbandIterationManager(BaseIterationManger):
experiment_ids = self.get_reduced_configs()
experiments = self.experiment_group.experiments.filter(id__in=experiment_ids)
self.create_iteration(experiment_ids=experiment_ids)
- self.experiment_group.refresh_from_db()
iteration_config = self.experiment_group.iteration_config
params_config = self.experiment_group.params_config
- resource_value = self.experiment_group.search_manager.get_resources_for_iteration(
+ n_resources = self.experiment_group.search_manager.get_resources_for_iteration(
iteration=iteration_config.iteration)
+ resource_value = self.experiment_group.search_manager.get_n_resources(
+ n_resources=n_resources, bracket_iteration=iteration_config.bracket_iteration
+ )
resource_name = params_config.hyperband.resource.name
resource_value = params_config.hyperband.resource.cast_value(resource_value)
