From 98cc7b5a9eaee2f9ed44e26c05d65c5693e32c4a Mon Sep 17 00:00:00 2001 From: Martin Man Date: Thu, 21 Jan 2021 09:23:50 -0800 Subject: [PATCH] Load the model path correctly (#6369) --- onnxruntime/python/tools/quantization/quantize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/quantization/quantize.py b/onnxruntime/python/tools/quantization/quantize.py index c7d190e8ea..0ab33442d2 100644 --- a/onnxruntime/python/tools/quantization/quantize.py +++ b/onnxruntime/python/tools/quantization/quantize.py @@ -48,7 +48,7 @@ def load_model(model_path: Path, optimize=True): onnx_model.replace_gemm_with_matmul() return onnx_model.model - return onnx.load(Path) + return onnx.load(Path(model_path)) def quantize(model,