diff --git a/show_cls.py b/show_cls.py
index 6d054265e..7e831fb7b 100644
--- a/show_cls.py
+++ b/show_cls.py
@@ -51,4 +51,4 @@
 
     pred_choice = pred.data.max(1)[1]
     correct = pred_choice.eq(target.data).cpu().sum()
-    print('i:%d  loss: %f accuracy: %f' %(i, loss.data[0], correct/float(32)))
+    print('i:%d  loss: %f accuracy: %f' %(i, loss.item(), correct/float(32)))