修复分类网络训练时tensor不连续报错

PyTorch/Compute-Vision/Accuracy_Validation/ResNet50/main_acc.py

@@ -501,7 +501,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].contiguous().view(-1).float().sum(0, keepdim=True)
             res.append(correct_k.mul_(100.0 / batch_size))
         return res