Skip to content

Commit de15b43

Browse files
committed
Fix a bug with accuracy retrieving from RealLabels
1 parent 6f43aeb commit de15b43

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/data/real_labels.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,6 @@ def add_result(self, output):
3737

3838
def get_accuracy(self, k=None):
3939
if k is None:
40-
return {k: float(np.mean(self.is_correct[k] for k in self.topk))}
40+
return {k: float(np.mean(self.is_correct[k])) * 100 for k in self.topk}
4141
else:
4242
return float(np.mean(self.is_correct[k])) * 100

0 commit comments

Comments
 (0)