瀏覽代碼

fix error for quantize inference model

FlyingQianMM 4 年之前
父節點
當前提交
4752562693
共有 1 個文件被更改,包括 6 次插入4 次删除
  1. 6 4
      paddlex/cv/models/slim/post_quantization.py

+ 6 - 4
paddlex/cv/models/slim/post_quantization.py

@@ -262,10 +262,12 @@ class PaddleXPostTrainingQuantization(PostTrainingQuantization):
             for var_name in self._quantized_act_var_name:
                 start = time.time()
                 sampling_data = []
-                filenames = [f for f in os.listdir(self._cache_dir) \
-                    if re.match(var_name + '_[0-9]+.npy', f)]
+                file_name = os.path.join(self._cache_dir, var_name)
+                cache_dir, var_name_ = os.path.split(file_name) 
+                filenames = [f for f in os.listdir(cache_dir) \
+                    if re.match(var_name_ + '_[0-9]+.npy', f)]
                 for filename in filenames:
-                    file_path = os.path.join(self._cache_dir, filename)
+                    file_path = os.path.join(cache_dir, filename)
                     sampling_data.append(np.load(file_path))
                     os.remove(file_path)
                 sampling_data = np.concatenate(sampling_data)
@@ -314,7 +316,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.makedir(save_dir)
                 np.save(save_path, var_tensor)
         else:
             for var_name in self._quantized_act_var_name: