use reshape instead of view in accuracy

pytorch · Sep 2, 2020 · 49e1a88 · 49e1a88
1 parent 4b119d7
commit 49e1a88
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/imagenet/main.py b/imagenet/main.py
@@ -420,7 +420,7 @@ def accuracy(output, target, topk=(1,)):
 
         res = []
         for k in topk:
-            correct_k = correct[:k].view(-1).float().sum(0, keepdim=True)
+            correct_k = correct[:k].reshape(-1).float().sum(0, keepdim=True)
             res.append(correct_k.mul_(100.0 / batch_size))
         return res