def hp_hyperband_start(self, experiment_group_id): experiment_group = get_valid_experiment_group(experiment_group_id=experiment_group_id) if not experiment_group: return should_retry = base.start_group_experiments(experiment_group=experiment_group) if should_retry: # Schedule another task self.retry(countdown=Intervals.EXPERIMENTS_SCHEDULER) hp_hyperband_iterate.delay(experiment_group_id=experiment_group_id)
def hp_random_search_start(self, experiment_group_id): experiment_group = get_running_experiment_group( experiment_group_id=experiment_group_id) if not experiment_group: return should_retry = base.start_group_experiments( experiment_group=experiment_group) if should_retry: # Schedule another task self.retry(countdown=Intervals.EXPERIMENTS_SCHEDULER) return experiment_group.set_status(status=ExperimentGroupLifeCycle.SUCCEEDED)
def hp_grid_search_start(self, experiment_group_id): experiment_group = get_running_experiment_group( experiment_group_id=experiment_group_id) if not experiment_group: return should_retry = base.start_group_experiments( experiment_group=experiment_group) if should_retry: # Schedule another task self.retry(countdown=Intervals.EXPERIMENTS_SCHEDULER) return base.check_group_experiments_finished(experiment_group_id)