From de15b438652119dfa3ba8c6593d916cf4c3edcf1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tymoteusz=20Wi=C5=9Bniewski?= Date: Fri, 4 Dec 2020 16:12:50 +0100 Subject: [PATCH] Fix a bug with accuracy retrieving from RealLabels --- timm/data/real_labels.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/data/real_labels.py b/timm/data/real_labels.py index a90be0c3..939c3486 100644 --- a/timm/data/real_labels.py +++ b/timm/data/real_labels.py @@ -37,6 +37,6 @@ class RealLabelsImagenet: def get_accuracy(self, k=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: return float(np.mean(self.is_correct[k])) * 100