Explorar o código

Merge pull request #510 from syyxsxx/develop

修改post_quantization.py
Jason %!s(int64=4) %!d(string=hai) anos
pai
achega
70fb70f85b
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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: