static init func

apache · Aug 23, 2021 · fea3791 · fea3791
1 parent 169e23c
commit fea3791
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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: