caffe2推理onnx报错IndexError: Input 475 is undefined

我的模型为预训练模型,格式为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就可以了。但是我们是预训练模型,也无法设置