reduce test time

This commit is contained in:
Chi Wang (MSR) 2021-02-05 23:35:47 -08:00
parent f956e957d4
commit cdca936b8b

View File

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