mxnet plugin

lanpa · Aug 29, 2017 · dbc5d78 · dbc5d78
1 parent 2c22da3
commit dbc5d78
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tensorboardX/x2num.py b/tensorboardX/x2num.py
@@ -13,6 +13,9 @@ def makenp(x, modality=None):
     if 'torch' in str(type(x)):
         return pytorch_np(x, modality)
 
+    if 'mxnet' in str(type(x)):
+        return mxnet_np(x, modality)
+
 def pytorch_np(x, modality):
     import torch
     if isinstance(x, torch.autograd.variable.Variable):
@@ -34,8 +37,9 @@ def theano_np(x):
 def caffe2_np(x):
     pass
 
-def mxnet_np(x):
-    pass
+def mxnet_np(x, modality):
+    return x.asnumpy()
+
 
 def chainer_np(x):
     import chainer