Browse Source

Merge pull request #510 from syyxsxx/develop

修改post_quantization.py
Jason 4 years ago
parent
commit
70fb70f85b
1 changed files with 1 additions and 1 deletions
  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")
                                          var_name + "_" + str(iter) + ".npy")
                 save_dir, file_name = os.path.split(save_path)
                 save_dir, file_name = os.path.split(save_path)
                 if not os.path.exists(save_dir):
                 if not os.path.exists(save_dir):
-                    os.mkdirs(save_dir)
+                    os.makedirs(save_dir)
                 np.save(save_path, var_tensor)
                 np.save(save_path, var_tensor)
         else:
         else:
             for var_name in self._quantized_act_var_name:
             for var_name in self._quantized_act_var_name: