Specify the agg_num param with BFT_args

alibaba · Mar 27, 2023 · 6e79d8a · 6e79d8a
1 parent 8657945
commit 6e79d8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/federatedscope/core/aggregators/krum_aggregator.py b/federatedscope/core/aggregators/krum_aggregator.py
@@ -14,7 +14,7 @@ class KrumAggregator(ClientsAvgAggregator):
     def __init__(self, model=None, device='cpu', config=None):
         super(KrumAggregator, self).__init__(model, device, config)
         self.byzantine_node_num = config.aggregator.byzantine_node_num
-        self.krum_agg_num = config.aggregator.krum.agg_num
+        self.krum_agg_num = config.aggregator.BFT_args.krum_agg_num
         assert 2 * self.byzantine_node_num + 2 < config.federate.client_num, \
             "it should be satisfied that 2*byzantine_node_num + 2 < client_num"