pre-commit

open-mmlab · Jun 29, 2023 · 86166b1 · 86166b1
1 parent 956f65c
commit 86166b1
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 37 deletions.
diff --git a/a.py b/a.py
diff --git a/mmpretrain/datasets/visdial.py b/mmpretrain/datasets/visdial.py
@@ -85,7 +85,7 @@ def load_data_list(self) -> List[dict]:
 
                 data_info['dialog_history'] = historys[dialog_id]
 
-                data_info['question'] = questions[question_id] + "?"
+                data_info['question'] = questions[question_id] + '?'
                 data_info['answer'] = answers[answer_id]
                 data_info['answer_options'] = answer_options
                 data_info['gt_answer_index'] = data_info[

diff --git a/mmpretrain/evaluation/metrics/visual_dialog.py b/mmpretrain/evaluation/metrics/visual_dialog.py
@@ -44,18 +44,16 @@ def process(self, data_batch, data_samples) -> None:
         """
         for sample in data_samples:
             answer_options = sample.get('answer_options')
-            
+
             G = torch.Generator()
             G.manual_seed(0)
             rank = 1 + torch.randperm(len(answer_options), generator=G)
-   
+
             pred_answer = sample.get('pred_answer')
-            
+
             if pred_answer in answer_options:
                 answer_index = answer_options.index(pred_answer)
                 rank[answer_index] = 1
-
-
 
             gt_index = sample.get('gt_answer_index')
             gt_rank = rank[gt_index]
@@ -79,7 +77,13 @@ def compute_metrics(self, results: List) -> dict:
         Mean = torch.tensor(results).float().mean()
         MRR = torch.tensor(results).reciprocal().mean()
 
-        metrics = {'R@1': R1.item(), 'R@5': R5.item(), 'R@10': R10.item(), 'Mean': Mean.item(), 'MRR': MRR.item()}
+        metrics = {
+            'R@1': R1.item(),
+            'R@5': R5.item(),
+            'R@10': R10.item(),
+            'Mean': Mean.item(),
+            'MRR': MRR.item()
+        }
         return metrics
 
     def _process_answer(self, answer) -> str: