From 99175b296a14c1b80596c1697af2e361b7ad7fa6 Mon Sep 17 00:00:00 2001 From: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com> Date: Tue, 4 Apr 2023 15:47:16 +0800 Subject: [PATCH] [Fix] fix acc evalustion wait for long (#1430) --- mmcls/evaluation/metrics/single_label.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmcls/evaluation/metrics/single_label.py b/mmcls/evaluation/metrics/single_label.py index 2611eed42..1a14adf83 100644 --- a/mmcls/evaluation/metrics/single_label.py +++ b/mmcls/evaluation/metrics/single_label.py @@ -188,7 +188,7 @@ class Accuracy(BaseMetric): pred_label['score'][ ..., self.index_candidates] = pred_label['score'][ ..., self.index_candidates] + 1.0 - result['pred_score'] = pred_label['score'] + result['pred_score'] = pred_label['score'].cpu() else: result['pred_label'] = pred_label['label'].cpu() result['gt_label'] = gt_label['label'].cpu()