Learn practical skills, build real-world projects, and advance your career
import torch
import torchvision
from torchvision.datasets import MNIST
dataset = MNIST(root='data/', download=True)
len(dataset)
60000
test_dataset = MNIST(root='data/', train=False)
len(test_dataset)
10000
dataset[0]
(<PIL.Image.Image image mode=L size=28x28 at 0x27CF16AC7B8>, tensor(5))