浏览代码

Merge pull request #510 from syyxsxx/develop

修改post_quantization.py
Jason 4 年之前
父节点
当前提交
70fb70f85b
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/cv/models/slim/post_quantization.py

+ 1 - 1
paddlex/cv/models/slim/post_quantization.py

@@ -314,7 +314,7 @@ class PaddleXPostTrainingQuantization(PostTrainingQuantization):
                                          var_name + "_" + str(iter) + ".npy")
                 save_dir, file_name = os.path.split(save_path)
                 if not os.path.exists(save_dir):
-                    os.mkdirs(save_dir)
+                    os.makedirs(save_dir)
                 np.save(save_path, var_tensor)
         else:
             for var_name in self._quantized_act_var_name: