santa icon indicating copy to clipboard operation
santa copied to clipboard

The train process error

Open YangBo0411 opened this issue 6 months ago • 3 comments

First of all, thank you for your outstanding work. When I was training, I encountered the following error: Traceback (most recent call last): File "train.py", line 18, in fix_b = torch.stack([test_loader_b.dataset[i]['A'] for i in range(opt.display_size)]).cuda() File "train.py", line 18, in fix_b = torch.stack([test_loader_b.dataset[i]['A'] for i in range(opt.display_size)]).cuda() File "E:\Github project\santa-main\data\single_dataset.py", line 33, in getitem A_path = self.A_paths[index] IndexError: list index out of range

The training command I used was ’python train.py --dataroot=datasets/multidataset --direction=AtoB --lambda_path=0.1 --tag=santa‘

There are four sub folders under the multidataset folder: trainA, trainB, testA, and testB. The images in the folder are unaligned.

Can you tell me what the problem is?

YangBo0411 avatar Jan 03 '24 13:01 YangBo0411