Torchvision dataset identify parameters

vision
1

Hi,

I’m a novice and am learning ML. I have loaded the pytorchvision dataset and trying to visualize the data through iterate method . How do I know that there are only parameters images and labels in the dataset? Where and how do I find this information? Any help is sincerely appreciated.

Download and load the test data

testset = datasets.MNIST(‘MNIST_data/’, download=True, train=False, transform=transform)
testloader = torch.utils.data.DataLoader(testset, batch_size=64, shuffle=True)
#Iterate
Testdataiter = iter(trainloader)
images,labels =dataiter.next()

2

You could have a look at the __getitem__ method of your Dataset to see all returned values.

1 Like