自学内容网 自学内容网

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

原文地址:https://blog.csdn.net/sdhdsf132452/article/details/143932458

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!