Skip to content

Underlying MXNet Model Extraction #85

Description

@joeddav

It would be beneficial to have a way to extract the underlying MXNet module, similar to how a user can call K.get_session() with the TensorFlow backend. The global _MODEL object is unaccessible via API, so there's no way I can see to access the MXNet module.

In my own fork I have added the following to mxnet_backend.py:

def get_mxnet_module():
    return _MODEL._mod

Please let me know if there's a better way to do it. If not, I'd be happy to submit a PR with the above as I believe that feature would be useful to others as well.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions