[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:
Servon
2020-10-21 14:56:20 +08:00
committed by GitHub
parent a075e37695
commit aeea168940
+7 -1
View File
@@ -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,