reshape

adobe · Sep 29, 2021 · b27a34a · b27a34a
1 parent d4bf038
commit b27a34a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -707,7 +707,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