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(
# Create ONNX Runtime session
options = SessionOptions()
session = InferenceSession(onnx_model.as_posix(), options)
session = InferenceSession(onnx_model.as_posix(), options, providers=["CPUExecutionProvider"])
# Compute outputs from the reference model
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