mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix topn metric view regression on PyTorch 1.7

Browse Source
This commit is contained in:
Ross Wightman 2020-10-29 14:04:15 -07:00
parent f944242cb0
commit 4a3df7842a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/utils/metrics.py
View File

@ -28,5 +28,5 @@ def accuracy(output, target, topk=(1,)):
batch_size = target.size(0) batch_size = target.size(0)
_, pred = output.topk(maxk, 1, True, True) _, pred = output.topk(maxk, 1, True, True)
pred = pred.t() pred = pred.t()
correct = pred.eq(target.view(1, -1).expand_as(pred)) correct = pred.eq(target.reshape(1, -1).expand_as(pred))
return [correct[:k].view(-1).float().sum(0) * 100. / batch_size for k in topk] return [correct[:k].reshape(-1).float().sum(0) * 100. / batch_size for k in topk]