Hello everyone.
I have 2 datasets for a classification problem. Each has labels of different sizes. I want to sample from both at the same time for every iteration of training. I could do something like:
train_dl = DataLoader(trainset_1+trainset_2, shuffle = True, batch_size=128, num_workers=4)
But the issue is the mismatch in dimensions of labels and the data loader can’t stack tensors of different dimensions.
Is there any workaround for this?