1. 情景介绍
我用pytorch训练的模型,将训练的模型转化为.onnx文件,然而,在模型验证的时候报错,错误栈如下:
ValidationError: Your model ir_version is higher than the checker's
局部代码如下:
import onnx
model = onnx.load("alexnet.onnx")
onnx.checker.check_model(model)
2. 分析问题并解决
原因:版本不匹配
原来用的onnx=1.4.1 需要升级一下onnx
pip install --upgrade onnx
升级之后就不会报错了。
每天进步一点,一起加油 !!!