Concat dataset

class ConcatDataset(torch.utils.data.Dataset):
    def __init__(self, *datasets):
        self.datasets = datasets

    def __getitem__(self, i):
        return tuple(d[i] for d in self.datasets)

    def __len__(self):
        return min(len(d) for d in self.datasets)

train_loader = torch.utils.data.DataLoader(
             ConcatDataset( # concat
                 datasets.ImageFolder(traindir_A),
                 datasets.ImageFolder(traindir_B)
             ),
             batch_size=args.batch_size, shuffle=True,
             num_workers=args.workers, pin_memory=True)

for i, (input, target) in enumerate(train_loader):
    ...

Source: discuss.pytorch.org

Similar codes

0 vote319 views1 answer
Python numpy bbox
Python
0 vote210 views1 answer
Django related_name abstract class
Python
0 vote224 views1 answer
No of words in a string in python
Python
45 votes231 views6 answers
How to replace nan values with 0 in pandas
Python
34 votes189 views2 answers
Python shuffle list with seed
Python
3 votes209 views1 answer
Python tkinter filedialog folder
Python
1 vote266 views1 answer
Python game over screen
Python
0 vote203 views1 answer
Numpy if zero is present
Python
0 vote190 views1 answer
Variable string in string python
Python
149 votes166 views20 answers
Reverse list python
Python