modeld: cleaner exit from onnx runner

old-commit-hash: 921a411da5
vw-mqb-aeb
Adeeb Shihadeh 3 years ago
parent cf0564382a
commit 3df7552b7b
  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