Skip to content
This repository has been archived by the owner on Aug 30, 2018. It is now read-only.

Allow to filter out inputs in caffe2_net_to_onnx_model #93

Open
dzhulgakov opened this issue Dec 13, 2017 · 0 comments
Open

Allow to filter out inputs in caffe2_net_to_onnx_model #93

dzhulgakov opened this issue Dec 13, 2017 · 0 comments

Comments

@dzhulgakov
Copy link
Member

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.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant