search

Dmmm1997 / SimVG

[NeurIPS2024] - SimVG: A Simple Framework for Visual Grounding with Decoupled Multi-modal Fusion
https://arxiv.org/abs/2409.17531
MIT License
44 stars 0 forks source link

An error occurred during training grefcoco datasets #3

Closed zhanght021 closed 1 week ago

zhanght021 commented 1 week ago

During the training process of the RefCOCO dataset, the following error will occur.

File "/home/XXXXX/SimVG-main/simvg/apis/train.py", line 49, in train_model for batch, inputs in enumerate(loader): File "/usr/local/python3.10.12/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 630, in next data = self._next_data() File "/usr/local/python3.10.12/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 673, in _next_data data = self._dataset_fetcher.fetch(index) # may raise StopIteration File "/usr/local/python3.10.12/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/usr/local/python3.10.12/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 52, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/XXXXX/SimVG-main/simvg/datasets/base.py", line 70, in getitem results = self.pipeline(results) File "/home/XXXXX/SimVG-main/simvg/datasets/pipelines/compose.py", line 22, in call results = transform(results) File "/home/XXXXX/SimVG-main/simvg/datasets/pipelines/loading.py", line 275, in call results = self._load_expression_tokenize_beit3(results) File "/home/XXXXX/SimVG-main/simvg/datasets/pipelines/loading.py", line 162, in _load_expression_tokenize_beit3 expression = expressions[self.random_ind] KeyError: 0

Dmmm1997 commented 1 week ago

During the training process of the RefCOCO dataset, the following error will occur.

File "/home/XXXXX/SimVG-main/simvg/apis/train.py", line 49, in train_model for batch, inputs in enumerate(loader): File "/usr/local/python3.10.12/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 630, in next data = self._next_data() File "/usr/local/python3.10.12/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 673, in _next_data data = self._dataset_fetcher.fetch(index) # may raise StopIteration File "/usr/local/python3.10.12/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/usr/local/python3.10.12/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 52, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/XXXXX/SimVG-main/simvg/datasets/base.py", line 70, in getitem results = self.pipeline(results) File "/home/XXXXX/SimVG-main/simvg/datasets/pipelines/compose.py", line 22, in call results = transform(results) File "/home/XXXXX/SimVG-main/simvg/datasets/pipelines/loading.py", line 275, in call results = self._load_expression_tokenize_beit3(results) File "/home/XXXXX/SimVG-main/simvg/datasets/pipelines/loading.py", line 162, in _load_expression_tokenize_beit3 expression = expressions[self.random_ind] KeyError: 0

I have update the grefcoco dataset link.

https://seunic-my.sharepoint.cn/:f:/g/personal/230238525_seu_edu_cn/EiX51qGWa9BBlt3fVRwgPhsBUxOIZ-yW3Hm7VXOQ3c2Ipw?e=M12MM6

plz re-download the grefcoco dataset and pull the newest version of the code.