瀏覽代碼

Update post_quantization.py

syyxsxx 4 年之前
父節點
當前提交
e98b106a7d
共有 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: