Class name fix for SecureCompressedAggregation

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,

src/decentralizepy/node/SecureCompressedAggregatopn.py → 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