From f7f793adc3bca8aec7b15ffbae07580310eaf429 Mon Sep 17 00:00:00 2001 From: "yuexiang.xyx" Date: Wed, 22 Feb 2023 11:31:31 +0800 Subject: [PATCH] bugfix for tree based model --- federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py | 1 + 1 file changed, 1 insertion(+) diff --git a/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py b/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py index 5e29e2e21..f29d29f16 100644 --- a/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py +++ b/federatedscope/vertical_fl/xgb_base/worker/train_wrapper.py @@ -46,6 +46,7 @@ def train(self, tree_num, node_num=None, training_info=None): def _check_eval_finish(self, tree_num): if self.eval_finish_flag: + self.eval_finish_flag = False if tree_num + 1 < self._cfg.model.num_of_trees: batch_index, feature_order_info = \ self.trainer.fetch_train_data()