Hi All,
I have a numpy array of modified MNIST, which has the dimensions of a working dataset (Nx28x28), and labels(N,)
I want to convert this to a PyTorch Dataset, so I did:
train = torch.utils.data.TensorDataset(img, labels.view(-1))
train_loader = torch.utils.data.DataLoader(train, batch_size=64, shuffle=False)

This causes an AssertionError for the dimensions when I try to optimize my model, and there’s zero transparency, since if I load the data from PyTorch my tensors and the train.dataset.train_data.dim() are the same.