From a36b91ce86ff105ac8d9e513aa88f42b85e33479 Mon Sep 17 00:00:00 2001 From: Wang Peng <798960736@qq.com> Date: Thu, 31 Aug 2023 22:17:05 +0800 Subject: [PATCH] Update vqa_gen_dataset.py, set max_tgt_length --- data/mm_data/vqa_gen_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/mm_data/vqa_gen_dataset.py b/data/mm_data/vqa_gen_dataset.py index f332c4d8..24b1633b 100644 --- a/data/mm_data/vqa_gen_dataset.py +++ b/data/mm_data/vqa_gen_dataset.py @@ -163,7 +163,7 @@ def __getitem__(self, index): ref_dict = {item.split('|!+')[1]: float(item.split('|!+')[0]) for item in ref.split('&&')} answer = max(ref_dict, key=ref_dict.get) conf = torch.tensor([ref_dict[answer]]) - tgt_item = self.encode_text(" {}".format(answer)) + tgt_item = self.encode_text(" {}".format(answer), length=self.max_tgt_length) if self.add_object and predict_objects is not None: predict_object_seq = ' '.join(predict_objects.strip().split('&&')[:self.max_object_length])