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 vote160 views2 answers
Createview django
Python Django
4 votes98 views2 answers
Change django admin title
Python
0 vote134 views1 answer
Django model specify table name
Python Django
0 vote105 views1 answer
Merge multile dict
Python
1 vote115 views1 answer
Import "django.core.urlresolvers" could not be resolved
Python
2 votes102 views1 answer
Filter function using lambda in python
Python
25 votes132 views6 answers
Import python file from another directory
Python
27 votes115 views3 answers
Pandas replce none with nan
Python
1 vote114 views1 answer
Two elements at a time in list comprehension
Python
0 vote116 views1 answer
Check pip installed packages inside virtualenv
Python