wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-11 12:43:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] Implement BOHB (#5382)

Browse Source
This commit is contained in:
Lisa Dunlap
2019-08-13 12:32:07 -07:00
committed by Richard Liaw
parent 79949fb8a0
commit b7d0733362
12 changed files with 519 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/schedulers/__init__.py
+2 -1
View File
@@ -4,6 +4,7 @@ from __future__ import print_function
from ray.tune.schedulers.trial_scheduler import TrialScheduler, FIFOScheduler
from ray.tune.schedulers.hyperband import HyperBandScheduler
from ray.tune.schedulers.hb_bohb import HyperBandForBOHB
from ray.tune.schedulers.async_hyperband import (AsyncHyperBandScheduler,
ASHAScheduler)
from ray.tune.schedulers.median_stopping_rule import MedianStoppingRule
@@ -12,5 +13,5 @@ from ray.tune.schedulers.pbt import PopulationBasedTraining
__all__ = [
"TrialScheduler", "HyperBandScheduler", "AsyncHyperBandScheduler",
"ASHAScheduler", "MedianStoppingRule", "FIFOScheduler",
"PopulationBasedTraining"
"PopulationBasedTraining", "HyperBandForBOHB"
]