Skip to content
Snippets Groups Projects
Commit 98b1561d authored by Rishi Sharma's avatar Rishi Sharma
Browse files

Reduce data saved

parent 3b26236f
No related branches found
No related tags found
No related merge requests found
......@@ -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(
......
......@@ -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(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment