Hi, so I am messing around with some datasets when suddenly I got a size error. This normally does not happen, so I check the shapes of the data and targets.
The shape of the MNIST targets is of size (batch_size), no classes anywhere.
x, y = iter(train_loader).next() print(y.shape)
Is this a bug with PyTorch, or am I missing something?