Fix a bug with accuracy retrieving from RealLabels

4 years ago · de15b43865
parent 6f43aeb252
commit de15b43865
1 changed files with 1 additions and 1 deletions
--- 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