Facing index error

[sandip761]

File "/home/sandip/edgeai-tensorlab/edgeai-torchvision/./references/edgeailite/main/pixel2pixel/train_segmentation_main.py", line 281, in run(args) File "/home/sandip/edgeai-tensorlab/edgeai-torchvision/./references/edgeailite/main/pixel2pixel/train_segmentation_main.py", line 276, in run main(arguments) File "/home/sandip/edgeai-tensorlab/edgeai-torchvision/./references/edgeailite/main/pixel2pixel/train_segmentation_main.py", line 123, in main train_pixel2pixel.main(arguemnts) File "/home/sandip/edgeai-tensorlab/edgeai-torchvision/references/edgeailite/edgeai_xvision/xengine/train_pixel2pixel.py", line 614, in main train(args, train_dataset, train_loader, model, optimizer, epoch, train_writer, scheduler, grad_scaler) File "/home/sandip/edgeai-tensorlab/edgeai-torchvision/references/edgeailite/edgeai_xvision/xengine/train_pixel2pixel.py", line 711, in train for iter_id, (inputs, targets) in enumerate(train_loader): File "/home/sandip/.pyenv/versions/torchvision/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 631, in next data = self._next_data() File "/home/sandip/.pyenv/versions/torchvision/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1346, in _next_data After flush 1 2812 549 return self._process_data(data) File "/home/sandip/.pyenv/versions/torchvision/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 1372, in _process_data data.reraise() File "/home/sandip/.pyenv/versions/torchvision/lib/python3.10/site-packages/torch/_utils.py", line 705, in reraise raise exception IndexError: Caught IndexError in DataLoader worker process 0. Original Traceback (most recent call last): File "/home/sandip/.pyenv/versions/torchvision/lib/python3.10/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop data = fetcher.fetch(index) # type: ignore[possibly-undefined] File "/home/sandip/.pyenv/versions/torchvision/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 "/home/sandip/.pyenv/versions/torchvision/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 51, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/sandip/edgeai-tensorlab/edgeai-torchvision/references/edgeailite/edgeai_xvision/xvision/datasets/cityscapes_plus.py", line 357, in getitem this_image_path = self.image_files[image_idx][index].rstrip() IndexError: list index out of range