在自己的数据集上训练出错

[aiwenzhu]

Traceback (most recent call last):
File "train_ssd.py", line 335, in
device=DEVICE, debug_steps=args.debug_steps, epoch=epoch)
File "train_ssd.py", line 117, in train
for i, data in enumerate(loader):
File "/home/ai/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 801, in next
return self._process_data(data)
File "/home/ai/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 846, in _process_data
data.reraise()
File "/home/ai/anaconda3/lib/python3.6/site-packages/torch/_utils.py", line 385, in reraise
raise self.exc_type(msg)
IndexError: Caught IndexError in DataLoader worker process 3.

IndexError: too many indices for array