onnx-caffe2 icon indicating copy to clipboard operation
onnx-caffe2 copied to clipboard

Allow to filter out inputs in caffe2_net_to_onnx_model

Open dzhulgakov opened this issue 7 years ago • 0 comments

Caffe2 nets usually have stubs for the inputs in a form of ConstantFill in the init_net. It causes caffe2_net_to_onnx_model to complain as it doesn't know how to handle this op.

Instead we should just allow user to specify the list of model inputs and filter them out from the init_net. Also, we can extract ValueInfo from the shape to make things nicer.

dzhulgakov avatar Dec 13 '17 06:12 dzhulgakov