Bootstrap

pytorch量化有关代码

get_onnx_info.py

from pprint import pprint
import onnxruntime

onnx_path = "lung_unet3d.onnx"
# onnx_path = "custompool/output.onnx"

provider = "CPUExecutionProvider"
onnx_session = onnxruntime.InferenceSession(onnx_path, providers
;