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 vote299 views2 answers
_xsrf argument missing from POST
Python
8 votes230 views2 answers
Path to create a text file in python
Python
1 vote276 views1 answer
Python get timestamp 2020-04-23T12:00:00Z
Python
0 vote313 views2 answers
Check anonim user django
Python Django
0 vote293 views1 answer
Block encoding request python
Python
2 votes276 views3 answers
How to sum digits of a number in python
Python
6 votes309 views2 answers
Python check if any elements in a list are different
Python
95 votes280 views6 answers
Pandas dataframe sort by column
Python
0 vote261 views1 answer
Subplots whitespace
Python
60 votes280 views2 answers
Sort_values
Python