Selaa lähdekoodia

增加模型验证结果日志

zhy 1 kuukausi sitten
vanhempi
commit
a3fddefa7f
1 muutettua tiedostoa jossa 1 lisäystä ja 0 poistoa
  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