diff --git a/example/auto_compression/README.md b/example/auto_compression/README.md index 8037dff152d7e2df4bb2f375f1216da18a1ee16d..b14b6593835537f7dce62e60bcd9c311eb434b76 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 5ca6616395aeb1991fafd4603f5ceb5e51e79849..1692b509510f289299d541cb18cdf10e65e7aee2 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,