我的模型为预训练模型,格式为tflite,然后转成了onnx格式的,并没有进行任何修改。
我在用caffe2推理onnx格式模型的时候,报错IndexError: Input 475 is undefined!。大家知道怎么解决吗
代码为
import onnx,caffe2
model=onnx.load(model_path)
prepared_backend=caffe2.python.onnx.backend.prepare(model)
执行到这里报错
Traceback (most recent call last):
File "test.py", line 15, in <module>
prepared_backend=caffe2.python.onnx.backend.prepare(model_1)
File "/home/eeodev/.local/lib/python3.6/site-packages/caffe2/python/onnx/backend.py", line 713, in prepare
init_net, predict_net = cls._onnx_model_to_caffe2_net(model, device, opset_version, False)
File "/home/eeodev/.local/lib/python3.6/site-packages/caffe2/python/onnx/backend.py", line 876, in _onnx_model_to_caffe2_net
onnx_model = onnx.utils.polish_model(onnx_model)
File "/usr/local/lib64/python3.6/site-packages/onnx/utils.py", line 24, in polish_model
model = onnx.optimizer.optimize(model)
File "/usr/local/lib64/python3.6/site-packages/onnx/optimizer.py", line 55, in optimize
optimized_model_str = C.optimize(model_str, passes)
IndexError: Input 475 is undefined!
我觉得应该是哪里没有设置,换了一个简单模型依然报这个错误。但是如果通过pytorch转换成onnx的时候设置torch.onnx.export()设置参数keep_initializers_as_inputs=true就可以了。但是我们是预训练模型,也无法设置