From de8ec78927c1ef4f85c133676db45bc479a0b3f0 Mon Sep 17 00:00:00 2001 From: Bartosz Smoczynski Date: Wed, 5 Nov 2025 13:28:40 +0100 Subject: [PATCH] Dump metrics as number not tensor --- timm/utils/metrics.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/timm/utils/metrics.py b/timm/utils/metrics.py index f313867789..558eac7b14 100644 --- a/timm/utils/metrics.py +++ b/timm/utils/metrics.py @@ -49,7 +49,7 @@ def final_accuracy(self, vrs): def _fa(vr): n_verified = round(vr * N) - return (n_verified + correct_sorted[n_verified:].sum()) / N + return (n_verified + correct_sorted[n_verified:].sum().item()) / N return [_fa(vr) for vr in vrs] @@ -67,8 +67,8 @@ def _afa(vr): afa_weights = torch.arange(1, N + 1) / n_verified return ( (n_verified - 1) / 2 - + (afa_weights[:n_verified] * correct_sorted[:n_verified]).sum() - + correct_sorted[n_verified:].sum() + + (afa_weights[:n_verified] * correct_sorted[:n_verified]).sum().item() + + correct_sorted[n_verified:].sum().item() ) / N return [_afa(vr) for vr in vrs]