From fea379141dc525793c91abff63860958630700fd Mon Sep 17 00:00:00 2001 From: shingjan Date: Mon, 23 Aug 2021 14:00:09 -0700 Subject: [PATCH] static init func --- python/tvm/autotvm/tuner/xgboost_cost_model.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/python/tvm/autotvm/tuner/xgboost_cost_model.py b/python/tvm/autotvm/tuner/xgboost_cost_model.py index 209838478d83..55196082c512 100644 --- a/python/tvm/autotvm/tuner/xgboost_cost_model.py +++ b/python/tvm/autotvm/tuner/xgboost_cost_model.py @@ -163,7 +163,7 @@ def _reset_pool(self, space, target, task): self.pool = PopenPoolExecutor( max_workers=self.num_threads, - initializer=self._initializer, + initializer=XGBoostCostModel._initializer, initargs=(space, target, task), ) @@ -332,8 +332,6 @@ def _get_feature(self, indexes): feas = pool.map_with_error_catching(self.feature_extract_func, need_extract) for i, fea in zip(need_extract, feas): fea_cache[i] = fea.value - print(fea) - print("_______________________\n") feature_len = None for idx in indexes: