Time for other ranks to save

pytorch · Feb 6, 2025 · 763f60e · 763f60e
1 parent a34a5fc
commit 763f60e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/ibm_rescaling/rescaling_demo.py b/examples/ibm_rescaling/rescaling_demo.py
@@ -140,6 +140,7 @@
     # Diag save
     os.makedirs(os.path.join(args.ckpt_path, "diag"))
     torch.save(data.state_dict(), os.path.join(args.ckpt_path, "diag", f"loader_state_{rank}.pth"))
+    time.sleep(10)
 
     # Perform data coverage check on rank 0 only
     if rank == 0: