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 a bug with accuracy retrieving from RealLabels

Browse Source
This commit is contained in:
Tymoteusz Wiśniewski 2020-12-04 16:12:50 +01:00
parent 6f43aeb252
commit de15b43865
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/data/real_labels.py
View File

@ -37,6 +37,6 @@ class RealLabelsImagenet:
def get_accuracy(self, k=None): def get_accuracy(self, k=None):
if k is None: if k is None:
return {k: float(np.mean(self.is_correct[k] for k in self.topk))} return {k: float(np.mean(self.is_correct[k])) * 100 for k in self.topk}
else: else:
return float(np.mean(self.is_correct[k])) * 100 return float(np.mean(self.is_correct[k])) * 100