RuntimeError: Expected index [1, 576, 384] to be smaller

[RishiGitH]

Not able togenerate warps , test_VITON.sh failling to run .I am getting this issue when i run sh test_VITON.sh . Is it due to image size ?

RuntimeError: Caught RuntimeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/opt/conda/envs/test/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop data = fetcher.fetch(index) File "/opt/conda/envs/test/lib/python3.8/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/envs/test/lib/python3.8/site-packages/torch/utils/data/utils/fetch.py", line 51, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/ubuntu/PF-AFN/PF-AFN_test/data/cp_dataset.py", line 172, in getitem parse_map = parse_map.scatter(0, parse, 1.0) RuntimeError: Expected index [1, 576, 384] to be smaller than self [20, 512, 384] apart from dimension 0