自定义数据集出错

``python3 train.py --datasets /home/skylark/Ultra-Light-Fast-Generic-Face-Detector-1MB/data/voc_dataset --validation_dataset /home/skylark/Ultra-Light-Fast-Generic-Face-Detector-1MB/data/voc_dataset 2025-02-28 22:51:54,488 - root - INFO - inpu size :320 priors nums:4420 2025-02-28 22:51:55,122 - root - INFO - Use Cuda. 2025-02-28 22:51:55,122 - root - INFO - Namespace(dataset_type='voc', datasets=['/home/skylark/Ultra-Light-Fast-Generic-Face-Detector-1MB/data/voc_dataset'], validation_dataset='/home/skylark/Ultra-Light-Fast-Generic-Face-Detector-1MB/data/voc_dataset', balance_data=False, net='RFB', freeze_base_net=False, freeze_net=False, lr=0.01, momentum=0.9, weight_decay=0.0005, gamma=0.1, base_net_lr=None, extra_layers_lr=None, base_net=None, pretrained_ssd=None, resume=None, scheduler='multi-step', milestones='80,100', t_max=120, batch_size=24, num_epochs=200, num_workers=4, validation_epochs=5, debug_steps=100, use_cuda=True, checkpoint_folder='models/', log_dir='./models/Ultra-Light(1MB)_&_Fast_Face_Detector/logs', cuda_index='0', power=2, overlap_threshold=0.35, optimizer_type='SGD', input_size=320) 2025-02-28 22:51:55,122 - root - INFO - Prepare training datasets. 2025-02-28 22:51:55,123 - root - INFO - No labels file, using default VOC classes. 2025-02-28 22:51:55,124 - root - INFO - Stored labels into file models/voc-model-labels.txt. 2025-02-28 22:51:55,124 - root - INFO - Train dataset size: 2582 2025-02-28 22:51:55,124 - root - INFO - Prepare Validation datasets. 2025-02-28 22:51:55,125 - root - INFO - No labels file, using default VOC classes. 2025-02-28 22:51:55,125 - root - INFO - validation dataset size: 277 2025-02-28 22:51:55,125 - root - INFO - Build network. 2025-02-28 22:51:55,266 - root - INFO - use gpu :[0] 2025-02-28 22:51:55,267 - root - INFO - Took 0.00 seconds to load the model. 2025-02-28 22:51:55,268 - root - INFO - Learning rate: 0.01, Base net learning rate: 0.01, Extra Layers learning rate: 0.01. 2025-02-28 22:51:55,268 - root - INFO - Uses MultiStepLR scheduler. 2025-02-28 22:51:55,268 - root - INFO - Start training from epoch 0. Traceback (most recent call last): File "/home/skylark/Ultra-Light-Fast-Generic-Face-Detector-1MB/train.py", line 357, in <module> train(train_loader, net, criterion, optimizer, File "/home/skylark/Ultra-Light-Fast-Generic-Face-Detector-1MB/train.py", line 136, in train for i, data in enumerate(loader): File "/home/skylark/.local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 631, in __next__ data = self._next_data() File "/home/skylark/.local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1346, in _next_data return self._process_data(data) File "/home/skylark/.local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1372, in _process_data data.reraise() File "/home/skylark/.local/lib/python3.10/site-packages/torch/_utils.py", line 722, in reraise raise exception IndexError: Caught IndexError …