Fix bug.

fxia22 · Dec 5, 2018 · f6eeca1 · f6eeca1
1 parent 6b8abaa
commit f6eeca1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/show_cls.py b/show_cls.py
@@ -50,4 +50,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.data.item(), correct / float(32)))