From 6f7698cb7c68ff73f66eda883fc127db951e8f14 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=9Clk=C3=BC=20Tuncer=20K=C3=BC=C3=A7=C3=BCkta=C5=9F?= <47727810+UlkuTuncerKucuktas@users.noreply.github.com> Date: Tue, 1 Dec 2020 09:25:18 +0300 Subject: [PATCH] Update accuracy.py (#104) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Ülkü Tuncer Küçüktaş --- mmcls/models/losses/accuracy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmcls/models/losses/accuracy.py b/mmcls/models/losses/accuracy.py index 2111ff432..d3fa57645 100644 --- a/mmcls/models/losses/accuracy.py +++ b/mmcls/models/losses/accuracy.py @@ -25,7 +25,7 @@ def accuracy_torch(pred, target, topk=1): correct = pred_label.eq(target.view(1, -1).expand_as(pred_label)) 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. / num)) return res