Reduce data saved

src/decentralizepy/sharing/PartialModel.py

@@ -70,8 +70,8 @@ class PartialModel(Sharing):
         self.total_meta = 0
 
         # Only save for 2 procs: Save space
-        if rank == 0 or rank == 1:
-            self.save_shared = True
+        if self.save_shared and not (rank == 0 or rank == 1):
+            self.save_shared = False
 
         if self.save_shared:
             self.folder_path = os.path.join(

src/decentralizepy/training/ChangeAccumulator.py

@@ -73,7 +73,7 @@ class ChangeAccumulator(Training):
             batch_size,
             shuffle,
         )
-        self.save_accumulated = conditional_value(save_accumulated, "", True)
+        self.save_accumulated = conditional_value(save_accumulated, "", False)
         self.communication_round = 0
         if self.save_accumulated:
             self.model_change_path = os.path.join(