Sets overwrite to true in dcp (#836)

Summary: Pull Request resolved: #836 Sets overwrite behavior to true in dcp Reviewed By: JKSenthil Differential Revision: D57349181 fbshipit-source-id: face5b5038f87b803d8c8ead04a812c8ff1c4d4b
pytorch · May 20, 2024 · 8430268 · 8430268
1 parent 38cf06c
commit 8430268
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchtnt/framework/callbacks/dcp_saver.py b/torchtnt/framework/callbacks/dcp_saver.py
@@ -353,6 +353,7 @@ def default_writer_options(self) -> Dict[str, Any]:
         dcp_options = {
             "thread_count": self._knob_options.max_per_rank_io_concurrency or 16,
             "sync_files": False,
+            "overwrite": True,
         }
         if dcp_options["thread_count"] > 1:
             dcp_options["single_file_per_rank"] = False