|
|
@@ -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
|
