mirror of
https://gitlab.alpinelinux.org/alpine/aports.git
synced 2025-08-06 05:47:13 +02:00
22 lines
793 B
Diff
22 lines
793 B
Diff
diff --git a/onnxruntime/python/tools/onnxruntime_test.py b/onnxruntime/python/tools/onnxruntime_test.py
|
|
index 11759f3..fbe7121 100644
|
|
--- a/onnxruntime/python/tools/onnxruntime_test.py
|
|
+++ b/onnxruntime/python/tools/onnxruntime_test.py
|
|
@@ -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()
|