[tune] All examples to use ConcurrencyLimiter (#10662)

Co-authored-by: Kai Fricke <kai@anyscale.com>
This commit is contained in:
Richard Liaw
2020-09-09 09:52:15 -07:00
committed by GitHub
parent cd5b99e5e0
commit 3501ea396c
14 changed files with 81 additions and 53 deletions
@@ -11,6 +11,7 @@ import time
import ray
from ray import tune
from ray.tune.suggest import ConcurrencyLimiter
from ray.tune.schedulers import AsyncHyperBandScheduler
from ray.tune.suggest.dragonfly import DragonflySearch
@@ -70,12 +71,14 @@ if __name__ == "__main__":
optimizer="bandit",
domain="euclidean",
# space=space, # If you want to set the space manually
metric="objective",
mode="max")
)
df_search = ConcurrencyLimiter(df_search, max_concurrent=4)
scheduler = AsyncHyperBandScheduler(metric="objective", mode="max")
scheduler = AsyncHyperBandScheduler()
tune.run(
objective,
metric="objective",
mode="max",
name="dragonfly_search",
search_alg=df_search,
scheduler=scheduler,