Explorar o código

增加模型验证结果日志

zhy hai 1 mes
pai
achega
a3fddefa7f
Modificáronse 1 ficheiros con 1 adicións e 0 borrados
  1. 1 0
      tests/verify_tool_accuracy_test.py

+ 1 - 0
tests/verify_tool_accuracy_test.py

@@ -68,6 +68,7 @@ if __name__ == '__main__':
         print(*onnx_files, sep='\n')
         for onnx_file in onnx_files:
             verify_result = verify_tool_mix.label_verification(onnx_file, args.framework, args.mode, args.model_type)
+            print(f"onnx_file: {onnx_file}, verify_result: {verify_result}")
             total += 1
             if str(verify_result) == args.except_result:
                 correct += 1