ray.tune.schedulers.HyperBandForBOHB#
- class ray.tune.schedulers.HyperBandForBOHB(time_attr: str = 'training_iteration', metric: str | None = None, mode: str | None = None, max_t: int = 81, reduction_factor: float = 3, stop_last_trials: bool = True)[source]#
Bases:
HyperBandScheduler
Extends HyperBand early stopping algorithm for BOHB.
This implementation removes the
HyperBandScheduler
pipelining. This class introduces key changes:1. Trials are now placed so that the bracket with the largest size is filled first.
2. Trials will be paused even if the bracket is not filled. This allows BOHB to insert new trials into the training.
See ray.tune.schedulers.HyperBandScheduler for parameter docstring.
Methods
Fair scheduling within iteration by completion percentage.
This provides a progress notification for the algorithm.
Adds new trial.
Cleans up trial info from bracket if trial completed early.
Cleans up trial info from bracket if trial errored early.
Notification when trial terminates.
If bracket is finished, all trials will be stopped.
Restore trial scheduler from checkpoint.
Save trial scheduler to a checkpoint
Attributes
Status for continuing trial execution
Status for pausing trial execution
Status for stopping trial execution