Просмотр исходного кода

修改屏蔽剪枝onnx文件问题

zhy 2 недель назад
Родитель
Сommit
c8b89811c6
1 измененных файлов с 1 добавлено и 1 удалено
  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: