Merge pull request #81 from lyakaap/master

fix isinstance() validation in pytorch 0.4
lanpa · Feb 8, 2018 · 79f1370 · 79f1370
2 parents 9cbfca8 + c5189bd
commit 79f1370
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':