fix gqa post-processing

juxieyou · May 25, 2023 · b38a7ac · b38a7ac
1 parent 794d352
commit b38a7ac
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lavis/tasks/vqa.py b/lavis/tasks/vqa.py
@@ -209,9 +209,9 @@ def _report_metrics(self, result_file, split):
  gt_ans = res["gt_ans"]
  pred = res["pred_ans"]
 
- if self.inference_method == "generate":
-  pred = vqa_tool.processPunctuation(pred)
-  pred = vqa_tool.processDigitArticle(pred)
+ # if self.inference_method == "generate":
+ pred = vqa_tool.processPunctuation(pred)
+ pred = vqa_tool.processDigitArticle(pred)
 
  vqa_acc = 1 if pred == gt_ans else 0