From e5f36761844e0185c26cd349c1d42b140561bd78 Mon Sep 17 00:00:00 2001
From: zhouzj <41366441+zzjjay@users.noreply.github.com>
Date: Tue, 28 Feb 2023 16:39:04 +0800
Subject: [PATCH] [cherry-pick] fix mkdirs on distributed training. (#1669)

---
 example/auto_compression/README.md        | 2 +-
 paddleslim/auto_compression/compressor.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/example/auto_compression/README.md b/example/auto_compression/README.md
index 8037dff1..b14b6593 100644
--- a/example/auto_compression/README.md
+++ b/example/auto_compression/README.md
@@ -255,7 +255,7 @@ ac.compress()
 ## 进阶使用
 
 - ACT可以自动处理常见的预测模型，如果有更特殊的改造需求，可以参考[ACT超参配置教程](./hyperparameter_tutorial.md)来进行单独配置压缩策略。
-- ACT接口各个参数详细含义可以参考 [ACT API文档](../docs/zh_cn/api_cn/static/auto-compression/auto_compression_api.rst)。
+- ACT接口各个参数详细含义可以参考 [ACT API文档](../../docs/zh_cn/api_cn/static/auto-compression/auto_compression_api.rst)。
 
 ## 社区交流
 
diff --git a/paddleslim/auto_compression/compressor.py b/paddleslim/auto_compression/compressor.py
index 5ca66163..1692b509 100644
--- a/paddleslim/auto_compression/compressor.py
+++ b/paddleslim/auto_compression/compressor.py
@@ -127,7 +127,7 @@ class AutoCompression:
 
         self.final_dir = save_dir
         if not os.path.exists(self.final_dir):
-            os.makedirs(self.final_dir)
+            os.makedirs(self.final_dir, exist_ok=True)
 
         # load config
         if isinstance(config, str):
@@ -263,7 +263,7 @@ class AutoCompression:
                     op.desc.infer_shape(block.desc)
 
         save_path = os.path.join(save_path, "infered_shape")
-        os.makedirs(save_path)
+        os.makedirs(save_path, exist_ok=True)
         paddle.static.save_inference_model(
             save_path,
             feed_vars,
-- 
GitLab