Error in Data Loader

[BaophanN]

Hi author, i try to reproduce the result on a single RTX 4090 GPU, following exactly the installation instruction. When I run:

python train.py --cfg_file ./cfgs/DA/phase1_vehicle_drone/source_only/pvrcnn_source.yaml

I got this error:

Traceback (most recent call last):                                                                                 | 0/1472 [00:00<?, ?it/s]
  File "/workspace/source/track5/tools/train.py", line 183, in <module>
    main()
  File "/workspace/source/track5/tools/train.py", line 156, in main
    train_model(
  File "/workspace/source/track5/tools/train_utils/train_utils.py", line 111, in train_model
    accumulated_iter = train_one_epoch(
  File "/workspace/source/track5/tools/train_utils/train_utils.py", line 25, in train_one_epoch
    batch = next(dataloader_iter)
  File "/opt/conda/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 630, in __next__
    data = self._next_data()
  File "/opt/conda/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1345, in _next_data
    return self._process_data(data)
  File "/opt/conda/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1371, in _process_data
    data.reraise()
  File "/opt/conda/lib/python3.10/site-packages/torch/_utils.py", line 694, in reraise
    raise exception
KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "/opt/conda/lib/python3.10/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop
    data = fetcher.fetch(index)
  File "/opt/conda/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/opt/conda/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 51, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/workspace/source/track5/tools/../pcdet/datasets/pi3det/pi3det_dataset.py", line 145, in __getitem__
    data_dict = self.prepare_data(data_dict=input_dict)
  File "/workspace/source/track5/tools/../pcdet/datasets/dataset.py", line 227, in prepare_data
    torch.from_numpy(data_dict['gt_boxes'][:, :7])).numpy().sum(axis=1)
KeyError: 'gt_boxes'

I followed exactly all steps of the installation process. But do not know why there is this error. Can you please help me fix this? Thank you

[AlanLiangC]

Hello, sorry for late response. Have u solved the error. If not, I need more details of the keys() of your data_dict.