[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.
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

View file

@ -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!"