|
@@ -100,7 +100,7 @@ if __name__ == '__main__':
|
|
|
|
|
|
# verify_result = verify_tool.label_verification(onnx_file)
|
|
|
# 调用验证工具进行标签验证
|
|
|
- verify_result = verify_tool_mix.label_verification(onnx_file, framework=args.framework, mode=args.mode, model_type=args.model_value)
|
|
|
+ verify_result = verify_tool_mix.label_verification(onnx_file, framework=args.framework, mode=args.mode, model_type=args.model_type)
|
|
|
total += 1
|
|
|
if str(verify_result) == args.except_result:
|
|
|
correct += 1
|