yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-09-26 16:51:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

reduce test time

Browse Source
This commit is contained in:
Chi Wang (MSR) 2021-02-05 23:35:47 -08:00
parent f956e957d4
commit cdca936b8b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_tune.py
View File

@ -57,7 +57,7 @@ def _test_xgboost(method='BlendSearch'):
} }
max_iter = 10 max_iter = 10
for num_samples in [256]: for num_samples in [256]:
time_budget_s = None time_budget_s = 60 #None
for n_cpu in [8]: for n_cpu in [8]:
start_time = time.time() start_time = time.time()
ray.init(num_cpus=n_cpu, num_gpus=0) ray.init(num_cpus=n_cpu, num_gpus=0)