Hey i just reviewed one part of code.

[genprokuror]

Hey i just reviewed one part of code.
Check out what's inside the training dataloader train_features_batch, train_labels_batch = next(iter(train_dataloader)) train_features_batch.shape, train_labels_batch.shape
I cant understand why it gives this when launching
(torch.Size([32, 1, 28, 28]), torch.Size([32]))
Why their sizes and values are not the same? Dont author assigned them together here? How it works?
train_features_batch, train_labels_batch = next(iter(train_dataloader))

Originally posted by @genprokuror in #811