saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-26 03:00:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Load the model path correctly (#6369)

Browse source
This commit is contained in:
Martin Man 2021-01-21 09:23:50 -08:00 committed by GitHub
parent 99a38f4023
commit 98cc7b5a9e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/quantization/quantize.py
View file

@ -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,