Merge pull request #541 from rayrayraykk/fix_demo

alibaba · Mar 15, 2023 · 883fc7b · 883fc7b
2 parents c776b99 + 4b1b351
commit 883fc7b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/benchmark/FedHPOBench/demo/base_fed_tabular_benchmark.py b/benchmark/FedHPOBench/demo/base_fed_tabular_benchmark.py
@@ -124,7 +124,7 @@ def _cost(self, configuration, fidelity):
             self.client_num * fidelity['sample_client'] * self.num_param /
             self.bandwidth['server_up'],
             self.num_param / self.bandwidth['client_down'])
-        return cmp_cost + cmm_cost
+        return (cmp_cost + cmm_cost) * fidelity['round']
 
     def _search(self, configuration, fidelity):
         # For configuration

diff --git a/benchmark/FedHPOBench/demo/femnist_surrogate_benchmark.py b/benchmark/FedHPOBench/demo/femnist_surrogate_benchmark.py
@@ -24,7 +24,7 @@ def __init__(self,
                     "/fedhpob_cnn_surrogate.zip"
         triplets = ('cnn', 'femnist', 'avg')
         client_num = 200
-        num_param = 871294
+        num_param = 6603902
         super(FENISTSurrogateFedHPOBench,
               self).__init__(data_path, model_path, data_url, model_url,
                              triplets, client_num, num_param, rng)

diff --git a/benchmark/FedHPOBench/demo/femnist_tabular_benchmark.py b/benchmark/FedHPOBench/demo/femnist_tabular_benchmark.py
@@ -37,7 +37,7 @@ def __init__(self,
               "/fedhpob_cnn_tabular.zip"
         triplets = ('cnn', 'femnist', 'avg')
         client_num = 200
-        num_param = 871294
+        num_param = 6603902
         super(FEMNISTTabularFedHPOBench, self).__init__(data_path,
                                                         url,
                                                         triplets,