Fix: PyTorch inference test
This commit is contained in:
@@ -47,6 +47,7 @@ def synthesize(
|
|||||||
|
|
||||||
# ロードされた InferenceSession の ExecutionProvider が一致するか確認
|
# ロードされた InferenceSession の ExecutionProvider が一致するか確認
|
||||||
# 一致しない場合、指定された ExecutionProvider で推論できない状態
|
# 一致しない場合、指定された ExecutionProvider で推論できない状態
|
||||||
|
if inference_type == "onnx":
|
||||||
assert model.onnx_session is not None
|
assert model.onnx_session is not None
|
||||||
assert model.onnx_session.get_providers()[0] == onnx_providers[0][0]
|
assert model.onnx_session.get_providers()[0] == onnx_providers[0][0]
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user