saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-18 21:21:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix: Add def main() in onnxruntime_test.py (#15208)

Browse source
This commit is contained in:
Leso_KN 2023-04-05 21:31:39 +02:00 committed by GitHub
parent ef11032c89
commit ea6b32fea8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/python/tools/onnxruntime_test.py
View file

@ -127,7 +127,7 @@ def run_model(
return 0, feeds, num_iters > 0 and outputs
if __name__ == "__main__":
def main():
parser = argparse.ArgumentParser(description="Simple ONNX Runtime Test Tool.")
parser.add_argument("model_path", help="model path")
parser.add_argument(
@ -155,3 +155,7 @@ if __name__ == "__main__":
args = parser.parse_args()
exit_code, _, _ = run_model(args.model_path, args.num_iters, args.debug, args.profile, args.symbolic_dims)
sys.exit(exit_code)
if __name__ == "__main__":
main()