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)!