Home Explore Help
Sign In
AIModelWatermark
/
model_watermark_detect
5
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

增加模型验证结果日志

zhy 1 month ago
parent
2450063d44
commit
a3fddefa7f
1 changed files with 1 additions and 0 deletions
Unified View Show Diff Stats
  1. 1 0
      tests/verify_tool_accuracy_test.py

+ 1 - 0
tests/verify_tool_accuracy_test.py
View File 

@@ -68,6 +68,7 @@ if __name__ == '__main__':
 
         print(*onnx_files, sep='\n')
 
         print(*onnx_files, sep='\n')
 
         for onnx_file in onnx_files:
 
         for onnx_file in onnx_files:
 
             verify_result = verify_tool_mix.label_verification(onnx_file, args.framework, args.mode, args.model_type)
 
             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
 
             total += 1
 
             if str(verify_result) == args.except_result:
 
             if str(verify_result) == args.except_result:
 
                 correct += 1
 
                 correct += 1