How to merge two torch.utils.data dataloaders with a single operation

[1chimaruGin (Ichimaru Gin)]

Train simultaneously on two datasets

I’d recommend creating a new dataset and concatenating the images there, so the copy will be done inside the worker processes: 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( datasets.Ima…