fix(python): do not fail if checkpoints path exist (#305)

Co-authored-by: root <root@bjlt-hg96.sy>
BaguaSys · Oct 18, 2021 · d600dcd · d600dcd
1 parent 6afcab3
commit d600dcd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bagua/torch_api/checkpoint/checkpointing.py b/bagua/torch_api/checkpoint/checkpointing.py
@@ -28,7 +28,7 @@ def _has_moe_layers(model: BaguaModule) -> Tuple[bool, int]:
 def _ensure_directory_exists(filename: str):
     dirname = os.path.dirname(filename)
     if not os.path.exists(dirname):
-        os.makedirs(dirname)
+        os.makedirs(dirname, exist_ok=True)
 
 
 def _get_optimizer_ckpt_name(