Error while trying python train.py --config configs/doc2doc.yaml (Custom Dataset)

[akshaydp1995]

Traceback (most recent call last): File "train.py", line 45, in train_display_images_b = torch.stack([train_loader_b.dataset[i] for i in range(display_size)]).cuda() File "train.py", line 45, in train_display_images_b = torch.stack([train_loader_b.dataset[i] for i in range(display_size)]).cuda() File "/content/MUNIT/data.py", line 119, in getitem path = self.imgs[index] IndexError: list index out of range

I've built on my own custom dataset called documents. Documents is a subfolder in datasets, containing 4 directories and 4 text files. Have made changes to .yaml file (changed last line to correspond to the correct dataset directory), and renamed as doc2doc.yaml

[APhun]

Hi, have you solved the problem? I had the same problem

[stillstream]

Hi, have you solved the problem? I had the same problem @APhun @akshaydp1995

[BassantSamy]

reducing the display_size from 8 to 3 did the trick for me