add mxnet module extraction support

resolves dmlc#85
joeddav · Oct 19, 2017 · 3a9a07b · 3a9a07b
1 parent a21823b
commit 3a9a07b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/keras/backend/mxnet_backend.py b/keras/backend/mxnet_backend.py
@@ -80,6 +80,10 @@ def set_model(model):
     _MODEL = model
 
 
+def get_mxnet_module():
+    return _MODEL._mod
+
+
 def clear_session():
     reset_uids()
     _EXECUTOR = None