Update vqa_gen_dataset.py, set max_tgt_length

OFA-Sys · Aug 31, 2023 · a36b91c · a36b91c
1 parent b358cd6
commit a36b91c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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])