以下是对 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 相关文档