hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 18:41:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[tune] Better ASHA defaults (#4623)

Browse source 
## 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.
This commit is contained in:
Richard Liaw 2019-04-15 01:45:43 -07:00 • committed by GitHub
parent f600591468
commit 776a7308c8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
python/ray/tune/schedulers/async_hyperband.py
View file

@ -18,7 +18,7 @@ class AsyncHyperBandScheduler(FIFOScheduler):
is when using multiple brackets, trial allocation to bracket is done is when using multiple brackets, trial allocation to bracket is done
randomly with over a softmax probability. randomly with over a softmax probability.
See https://openreview.net/forum?id=S1Y7OOlRZ See https://arxiv.org/abs/1810.05934
Args: Args:
time_attr (str): A training result attr to use for comparing time. time_attr (str): A training result attr to use for comparing time.
@ -43,8 +43,8 @@ class AsyncHyperBandScheduler(FIFOScheduler):
reward_attr="episode_reward_mean", reward_attr="episode_reward_mean",
max_t=100, max_t=100,
grace_period=10, grace_period=10,
reduction_factor=3, reduction_factor=4,
brackets=3): brackets=1):
assert max_t > 0, "Max (time_attr) not valid!" assert max_t > 0, "Max (time_attr) not valid!"
assert max_t >= grace_period, "grace_period must be <= max_t!" assert max_t >= grace_period, "grace_period must be <= max_t!"
assert grace_period > 0, "grace_period must be positive!" assert grace_period > 0, "grace_period must be positive!"