"docs/source/vscode:/vscode.git/clone" did not exist on "7f5ad6c35b2b2b8a4710f87b868a7eb17c14a022"
Unverified Commit 95a75a71 authored by Ye Wang's avatar Ye Wang Committed by GitHub
Browse files

Specify providers explicitly in ORT session initialization (#15235)



* Specify providers explicitly in ORT session initialization
Co-authored-by: default avatarUbuntu <wy@linux-v100.aidmrjtolptuzevavgwhrapqcd.jx.internal.cloudapp.net>
parent 833635e2
...@@ -164,7 +164,7 @@ def validate_model_outputs( ...@@ -164,7 +164,7 @@ def validate_model_outputs(
# Create ONNX Runtime session # Create ONNX Runtime session
options = SessionOptions() options = SessionOptions()
session = InferenceSession(onnx_model.as_posix(), options) session = InferenceSession(onnx_model.as_posix(), options, providers=["CPUExecutionProvider"])
# Compute outputs from the reference model # Compute outputs from the reference model
ref_outputs = reference_model(**reference_model_inputs) ref_outputs = reference_model(**reference_model_inputs)
......
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