[Fix] fix acc evalustion wait for long (#1430)

2023-04-04 15:47:16 +08:00 · 2023-04-04 15:47:16 +08:00 · 99175b296a
parent 34a05f969f
commit 99175b296a
1 changed files with 1 additions and 1 deletions
--- 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()