From d667351eac57da2b07a50c07482652103a7839ee Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Thu, 19 Aug 2021 14:18:53 -0700
Subject: [PATCH] Tweak accuracy topk safety. Fix #807

---
 timm/utils/metrics.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/timm/utils/metrics.py b/timm/utils/metrics.py
index 4f5d95a1..9fdbe13e 100644
--- a/timm/utils/metrics.py
+++ b/timm/utils/metrics.py
@@ -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]