Skip to content
This repository was archived by the owner on Aug 30, 2018. It is now read-only.
This repository was 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

@dzhulgakov

Description

@dzhulgakov

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions