mirrors
/
mmclassification
mirror of https://github.com/open-mmlab/mmclassification.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

pull/1456/head
Ezra-Yu 2023-04-04 15:47:16 +08:00 committed by GitHub
parent 34a05f969f
commit 99175b296a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmcls/evaluation/metrics/single_label.py
View File

@ -188,7 +188,7 @@ class Accuracy(BaseMetric):
pred_label['score'][ pred_label['score'][
..., self.index_candidates] = pred_label['score'][ ..., self.index_candidates] = pred_label['score'][
..., self.index_candidates] + 1.0 ..., self.index_candidates] + 1.0
result['pred_score'] = pred_label['score'] result['pred_score'] = pred_label['score'].cpu()
else: else:
result['pred_label'] = pred_label['label'].cpu() result['pred_label'] = pred_label['label'].cpu()
result['gt_label'] = gt_label['label'].cpu() result['gt_label'] = gt_label['label'].cpu()