zjt/SAMapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix lint.

Browse Source
This commit is contained in:
Eric Mintun 2023-04-15 10:34:28 -07:00
parent 3518c86b78
commit 02acff7859
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/export_onnx_model.py
View File

@ -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.")