fix isinstance() validation in pytorch

jx57 · Feb 8, 2018 · c5189bd · c5189bd
1 parent 9cbfca8
commit c5189bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorboardX/x2num.py b/tensorboardX/x2num.py
@@ -21,7 +21,7 @@ def makenp(x, modality=None):
 
 def pytorch_np(x, modality):
     import torch
-    if isinstance(x, torch.autograd.variable.Variable):
+    if isinstance(x, torch.autograd.Variable):
         x = x.data
     x = x.cpu().numpy()
     if modality == 'IMG':