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

2 votes265 views2 answers
Seaborn boxplot multiple columns
Python
14 votes165 views5 answers
Discord py get user by id
Python
9 votes144 views4 answers
Python get dates between two dates
Python
0 vote155 views2 answers
Plt.text background alpha
Python
0 vote163 views1 answer
Python interpreter pip
Python
23 votes185 views6 answers
Decorator python
Python
5 votes206 views2 answers
Get pyhton file path python
Python
0 vote189 views1 answer
Modules django
Python Django
3 votes159 views1 answer
Get biggest value in array python3
Python
0 vote160 views3 answers
Python nc group variables
Python