Skip to content

GitLab

Commit 02acff78 authored by Eric Mintun's avatar Eric Mintun
Browse files

Fix lint.

parent 3518c86b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
scripts/export_onnx_model.py
View file @ 02acff78
...@@ -161,7 +161,7 @@ def run_export( ...@@ -161,7 +161,7 @@ def run_export(
if onnxruntime_exists: if onnxruntime_exists:
ort_inputs = {k: to_numpy(v) for k, v in dummy_inputs.items()} ort_inputs = {k: to_numpy(v) for k, v in dummy_inputs.items()}
# set cpu provider default # set cpu provider default
providers = ['CPUExecutionProvider'] providers = ["CPUExecutionProvider"]
ort_session = onnxruntime.InferenceSession(output, providers=providers) ort_session = onnxruntime.InferenceSession(output, providers=providers)
_ = ort_session.run(None, ort_inputs) _ = ort_session.run(None, ort_inputs)
print("Model has successfully been run with ONNXRuntime.") print("Model has successfully been run with ONNXRuntime.")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment