Explorar o código

修改屏蔽剪枝onnx文件问题

zhy hai 2 semanas
pai
achega
c8b89811c6
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      tests/verify_tool_accuracy_test.py

+ 1 - 1
tests/verify_tool_accuracy_test.py

@@ -91,7 +91,7 @@ if __name__ == '__main__':
         if args.model_file_filter:
             onnx_files = [item for item in onnx_files if args.model_file_filter in item]
         else:
-            onnx_files = [item for item in onnx_files if "pruned" not in item]
+            onnx_files = [item for item in onnx_files]
         print(f"model_name: {model_dir}\nonnx_files:")
         print(*onnx_files, sep='\n')
         for onnx_file in onnx_files: