Update run.py (#7)

Author: sxj731533730

run.py

@@ -83,7 +83,7 @@ def correct(output, target, topk=(1,)):
 
     res = []
     for k in topk:
-        correct_k = correct[:k].view(-1).float().sum(0).item()
+        correct_k = correct[:k].reshape(-1).float().sum(0).item()
         res.append(correct_k)
     return res