diff --git a/eval/testing_secure.py b/eval/testing_secure.py index 6d03d318d0f48865c9d64c19c624a64660e11012..2f2ad656149625f212029c8109e10f1ef8567e16 100644 --- a/eval/testing_secure.py +++ b/eval/testing_secure.py @@ -8,7 +8,7 @@ from torch import multiprocessing as mp from decentralizepy import utils from decentralizepy.graphs.Graph import Graph from decentralizepy.mappings.Linear import Linear -from decentralizepy.node.SecureCompressedAggregatopn import SecureCompressedAggregatopn +from decentralizepy.node.SecureCompressedAggregation import SecureCompressedAggregation def read_ini(file_path): @@ -54,7 +54,7 @@ if __name__ == "__main__": for r in range(procs_per_machine): processes.append( mp.Process( - target=SecureCompressedAggregatopn, + target=SecureCompressedAggregation, args=[ r, m_id, diff --git a/src/decentralizepy/node/SecureCompressedAggregatopn.py b/src/decentralizepy/node/SecureCompressedAggregation.py similarity index 98% rename from src/decentralizepy/node/SecureCompressedAggregatopn.py rename to src/decentralizepy/node/SecureCompressedAggregation.py index 465be84667608b8944c0b46c060ba68b07c79204..76166949ab0b86711a6ea2812cb9b8f9c7fd2caa 100644 --- a/src/decentralizepy/node/SecureCompressedAggregatopn.py +++ b/src/decentralizepy/node/SecureCompressedAggregation.py @@ -84,7 +84,7 @@ def temp_seed(seed): finally: np.random.set_state(state) -class SecureCompressedAggregatopn(DPSGDNode): +class SecureCompressedAggregation(DPSGDNode): """ This class defines the node for secure compressed aggregation @@ -226,7 +226,8 @@ class SecureCompressedAggregatopn(DPSGDNode): def generate_mask(self, seed, size): with temp_seed(seed): - return torch.Tensor(np.random.uniform(1, 10, size=size)) + # Figure out best distribution to add + return torch.Tensor(np.random.normal(0, 100000, size=size)) def run(self): """ @@ -244,7 +245,6 @@ class SecureCompressedAggregatopn(DPSGDNode): for iteration in range(self.iterations): logging.info("Starting training iteration: %d", iteration) - print("Starting training iteration:", iteration) rounds_to_train_evaluate -= 1 rounds_to_test -= 1