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.
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
: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.