From 776a7308c846482596f25038fc54112cdbd0af02 Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Mon, 15 Apr 2019 01:45:43 -0700 Subject: [PATCH] [tune] Better ASHA defaults (#4623) ## What do these changes do? Sets ASHA defaults to paper defaults. ## Related issue number ## Linter - [ ] I've run `scripts/format.sh` to lint the changes in this PR. --- python/ray/tune/schedulers/async_hyperband.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/ray/tune/schedulers/async_hyperband.py b/python/ray/tune/schedulers/async_hyperband.py index 55bf169f0..da06bfdbb 100644 --- a/python/ray/tune/schedulers/async_hyperband.py +++ b/python/ray/tune/schedulers/async_hyperband.py @@ -18,7 +18,7 @@ class AsyncHyperBandScheduler(FIFOScheduler): is when using multiple brackets, trial allocation to bracket is done randomly with over a softmax probability. - See https://openreview.net/forum?id=S1Y7OOlRZ + See https://arxiv.org/abs/1810.05934 Args: time_attr (str): A training result attr to use for comparing time. @@ -43,8 +43,8 @@ class AsyncHyperBandScheduler(FIFOScheduler): reward_attr="episode_reward_mean", max_t=100, grace_period=10, - reduction_factor=3, - brackets=3): + reduction_factor=4, + brackets=1): assert max_t > 0, "Max (time_attr) not valid!" assert max_t >= grace_period, "grace_period must be <= max_t!" assert grace_period > 0, "grace_period must be positive!"