modify cfgs of data

alibaba · Mar 27, 2023 · 2d02565 · 2d02565
1 parent 6e79d8a
commit 2d02565
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_robust_aggregators.py b/tests/test_robust_aggregators.py
@@ -20,6 +20,7 @@ def set_config_multikrum(self, cfg):
         cfg.merge_from_file(
             'federatedscope/cv/baseline/fedavg_convnet2_on_femnist.yaml')
 
+        cfg.data.root = 'test_data/'
         cfg.federate.client_num = 50
         cfg.federate.total_round_num = 10
         cfg.aggregator.byzantine_node_num = 10
@@ -38,6 +39,7 @@ def set_config_median(self, cfg):
         import torch
         cfg.merge_from_file(
             'federatedscope/cv/baseline/fedavg_convnet2_on_femnist.yaml')
+        cfg.data.root = 'test_data/'
         cfg.federate.client_num = 50
         cfg.federate.total_round_num = 10
 
@@ -57,6 +59,7 @@ def set_config_trimmedmean(self, cfg):
         import torch
         cfg.merge_from_file(
             'federatedscope/cv/baseline/fedavg_convnet2_on_femnist.yaml')
+        cfg.data.root = 'test_data/'
         cfg.federate.client_num = 50
         cfg.federate.total_round_num = 10
 
@@ -78,6 +81,7 @@ def set_config_bulyan(self, cfg):
         cfg.merge_from_file(
             'federatedscope/cv/baseline/fedavg_convnet2_on_femnist.yaml')
 
+        cfg.data.root = 'test_data/'
         cfg.federate.client_num = 50
         cfg.federate.total_round_num = 10
 
@@ -97,9 +101,9 @@ def set_config_normbounding(self, cfg):
         cfg.merge_from_file(
             'federatedscope/cv/baseline/fedavg_convnet2_on_femnist.yaml')
 
+        cfg.data.root = 'test_data/'
         cfg.federate.client_num = 50
         cfg.federate.total_round_num = 10
-
         cfg.aggregator.robust_rule = 'normbounding'
         cfg.aggregator.BFT_args.normbounding_norm_bound = 5
         cfg.aggregator.byzantine_node_num = 10