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