Fix topn metric view regression on PyTorch 1.7

4 years ago · 4a3df7842a
parent f944242cb0
commit 4a3df7842a
1 changed files with 2 additions and 2 deletions
--- a/timm/utils/metrics.py
+++ b/timm/utils/metrics.py
@ -28,5 +28,5 @@ def accuracy(output, target, topk=(1,)):
    batch_size = target.size(0)
    _, pred = output.topk(maxk, 1, True, True)
    pred = pred.t()
-    correct = pred.eq(target.view(1, -1).expand_as(pred))
-    return [correct[:k].view(-1).float().sum(0) * 100. / batch_size for k in topk]
+    correct = pred.eq(target.reshape(1, -1).expand_as(pred))
+    return [correct[:k].reshape(-1).float().sum(0) * 100. / batch_size for k in topk]