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

[tune] PB2 (#11466)

Browse Source 
Co-authored-by: Sumanth Ratna <sumanthratna@gmail.com>
Co-authored-by: Amog Kamsetty <amogkamsetty@yahoo.com>
Co-authored-by: Amog Kamsetty <amogkam@users.noreply.github.com>
Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
Jack Parker-Holder
2020-10-27 01:03:21 -07:00
committed by GitHub
co-authored by Sumanth Ratna Amog Kamsetty Amog Kamsetty Richard Liaw
parent 349c3ec86b
commit e7aafd7d24
12 changed files with 859 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/schedulers/__init__.py
+3 -1
View File
@@ -6,6 +6,7 @@ from ray.tune.schedulers.async_hyperband import (AsyncHyperBandScheduler,
from ray.tune.schedulers.median_stopping_rule import MedianStoppingRule
from ray.tune.schedulers.pbt import (PopulationBasedTraining,
PopulationBasedTrainingReplay)
from ray.tune.schedulers.pb2 import PB2
def create_scheduler(
@@ -37,6 +38,7 @@ def create_scheduler(
"hb_bohb": HyperBandForBOHB,
"pbt": PopulationBasedTraining,
"pbt_replay": PopulationBasedTrainingReplay,
"pb2": PB2,
}
scheduler = scheduler.lower()
if scheduler not in SCHEDULER_IMPORT:
@@ -52,5 +54,5 @@ __all__ = [
"TrialScheduler", "HyperBandScheduler", "AsyncHyperBandScheduler",
"ASHAScheduler", "MedianStoppingRule", "FIFOScheduler",
"PopulationBasedTraining", "PopulationBasedTrainingReplay",
"HyperBandForBOHB"
"HyperBandForBOHB", "PB2"
]