OFA
OFA copied to clipboard
"IndexError: list index out of range" on multiple GPU
Hi, when I try to run evaluate_vqa_beam.sh to generate test_dev's answers on multiple GPU, there always occurs error.
File "OFA/data/mm_data/vqa_gen_dataset.py", line 149, in getitem
item = self.dataset[index]
File "OFA/data/file_dataset.py", line 106, in getitem
column_l = [dtype(column_l[col_id]) for col_id, dtype in zip(self.selected_col_ids, self.dtypes)]
File "OFA/data/file_dataset.py", line 106, in
But when I run single GPUS_PER_NODE=1, the problem does not occur and could generate answers successfully.
By the way, the vqa_val.tsv which download from #68 could run evaluate_vqa_beam.sh on multiple GPU, but my fine only could run on single GPU.
Hi, I still met IndexError
while fine-tuning at this line