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

Tweak accuracy topk safety. Fix #807

Browse Source
This commit is contained in:
Ross Wightman 2021-08-19 14:18:53 -07:00
parent 35c9740826
commit d667351eac
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
timm/utils/metrics.py
View File

@ -2,7 +2,6 @@
Hacked together by / Copyright 2020 Ross Wightman
"""
import torch
class AverageMeter:
@ -30,7 +29,4 @@ def accuracy(output, target, topk=(1,)):
_, pred = output.topk(maxk, 1, True, True)
pred = pred.t()
correct = pred.eq(target.reshape(1, -1).expand_as(pred))
return [
correct[:k].reshape(-1).float().sum(0) * 100. / batch_size
if k <= maxk else torch.tensor(100.) for k in topk
]
return [correct[:min(k, maxk)].reshape(-1).float().sum(0) * 100. / batch_size for k in topk]