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