Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career

Image classification with Pytorch logestic regression

import torch
import torchvision
from torchvision.datasets import MNIST
ds = MNIST(root='\data', download=True)
len(ds)
60000
ds[0]
(<PIL.Image.Image image mode=L size=28x28 at 0x7F7C1C5DAEF0>, 5)