Update n_trials

rapidsai · Sep 23, 2023 · b759e83 · b759e83
1 parent 8ade306
commit b759e83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/examples/xgboost-rf-gpu-cpu-benchmark/hpo.py b/source/examples/xgboost-rf-gpu-cpu-benchmark/hpo.py
@@ -151,7 +151,7 @@ def main(args):
             threads_per_worker=args.threads_per_worker,
         )
     n_workers = len(cluster.workers)
-    n_trials = 8
+    n_trials = 100
 
     dataset = None
     # For CPU target, load the data matrix once and shared it among workers