【如何检查 ONNX 模型是否正确?】onnx.checker.check_model 用法详解

54 篇文章 2 订阅
22 篇文章 0 订阅

以下是对 onnx.checker.check_model 函数文档的翻译:

onnx.checker.check_model

检查模型的一致性,即模型在结构、格式和配置方面的正确性和完整性。

如果模型的 ir_version 设置不正确或高于检查器的 ir_version,或者模型在 metadata_props 中有重复键,将引发异常。

如果 IR version >= 3,模型必须指定 opset_import。如果 IR version < 3,模型不能指定 opset_import

参数:
  • model:要检查的模型。如果模型是一个路径,函数会首先检查模型路径。如果模型的字节大小超过2GB,应使用模型路径来调用该函数。
  • full_check:如果为 True,函数还会运行形状推断检查。
  • skip_opset_compatibility_check:如果为 True,函数将跳过算子集兼容性检查。
  • check_custom_domain:如果为 True,函数将检查所有域。否则,仅检查内置域。

示例代码

import onnx

# 检查 onnx 计算图
onnx_model = onnx.load("/path/to/yourmodel.onnx")
# 检查模型
try:
    onnx.checker.check_model(onnx_model)
    print("Model check passed.")
except Exception as e:
    print(f"Model check failed: {e}")

如果没问题,就会打印出 “Model check passed.”

参考内容:
[1] onnx.checker.check_model 相关文档

  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值