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 vote306 views2 answers
How to terminate subprocess.call in python
Python
1 vote321 views1 answer
How to give multiple option to the user and ask the same question again and again until the user tells one of the options
Python
0 vote293 views1 answer
Find total no of true in a list in python
Python
0 vote329 views1 answer
No module named googlesearch
Python
3 votes350 views3 answers
Python thread with parameters
Python
23 votes309 views2 answers
Converting month number to month name python
Python
0 vote324 views1 answer
Django check if get parameter exists
Python Django
4 votes309 views1 answer
Pyvenv.cfg file download
Python
0 vote322 views1 answer
Pyspark concat columns
Python
1 vote311 views1 answer
.pyc
Python