modeld: cleaner exit from onnx runner

pull/24728/head
Adeeb Shihadeh 3 years ago
parent dc4d4f875b
commit 921a411da5
  1. 11
      selfdrive/modeld/runners/onnx_runner.py

@ -59,7 +59,10 @@ if __name__ == "__main__":
options.graph_optimization_level = ort.GraphOptimizationLevel.ORT_ENABLE_ALL
provider = 'CPUExecutionProvider'
print("Onnx selected provider: ", [provider], file=sys.stderr)
ort_session = ort.InferenceSession(sys.argv[1], options, providers=[provider])
print("Onnx using ", ort_session.get_providers(), file=sys.stderr)
run_loop(ort_session)
try:
print("Onnx selected provider: ", [provider], file=sys.stderr)
ort_session = ort.InferenceSession(sys.argv[1], options, providers=[provider])
print("Onnx using ", ort_session.get_providers(), file=sys.stderr)
run_loop(ort_session)
except KeyboardInterrupt:
pass

Loading…
Cancel
Save