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)[源代码]#

基类:HyperBandScheduler

扩展了BOHB的超带早期停止算法。

此实现移除了 HyperBandScheduler 流水线。此类引入了关键更改:

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.

参见 ray.tune.schedulers.HyperBandScheduler 的参数文档字符串。

方法

choose_trial_to_run

按完成百分比在迭代内进行公平调度。

debug_string

这为算法提供了一个进度通知。

on_trial_add

添加新试验。

on_trial_complete

如果试验提前完成,则从括号中清除试验信息。

on_trial_error

如果试验早期出错,则从括号中清除试验信息。

on_trial_remove

试用期结束时的通知。

on_trial_result

如果括号结束,所有试验将被停止。

restore

从检查点恢复试用调度器。

save

将试验调度器保存到检查点

属性

CONTINUE

继续试验执行的状态

NOOP

PAUSE

暂停试验执行的状态

STOP

停止试验执行的状态

metric

supports_buffered_results