[AutoTVM] Update XGBoost verbosity option (#5649)

apache · May 22, 2020 · b26f988 · b26f988
1 parent dbb8be7
commit b26f988
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/autotvm/tuner/xgboost_cost_model.py b/python/tvm/autotvm/tuner/xgboost_cost_model.py
@@ -118,7 +118,7 @@ def __init__(self, task, feature_type, loss_type, num_threads=None, log_interval
         else:
             raise RuntimeError("Invalid loss type: " + loss_type)
 
-        self.xgb_params['silent'] = 1
+        self.xgb_params['verbosity'] = 0
         if num_threads:
             self.xgb_params['nthread'] = num_threads
         self.bst = None