mirror of
https://github.com/wassname/ray.git
synced 2026-07-07 04:30:30 +08:00
[tune] Update for ZOOpt (#11491)
Co-authored-by: Servon <zewen.li@polixir.ai> Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
@@ -43,12 +43,18 @@ class ZOOptSearch(Searcher):
|
||||
"height": tune.uniform(-10, 10)
|
||||
}
|
||||
|
||||
zoopt_search_config = {
|
||||
"parallel_num": 8, # how many workers to parallel
|
||||
}
|
||||
|
||||
zoopt_search = ZOOptSearch(
|
||||
algo="Asracos", # only support Asracos currently
|
||||
budget=20, # must match `num_samples` in `tune.run()`.
|
||||
dim_dict=dim_dict,
|
||||
metric="mean_loss",
|
||||
mode="min")
|
||||
mode="min",
|
||||
**zoopt_search_config
|
||||
)
|
||||
|
||||
tune.run(my_objective,
|
||||
config=config,
|
||||
|
||||
Reference in New Issue
Block a user