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

Image Classification

Dataset : CIFAR

import torch
import torchvision
import torchvision.transforms as transforms
dataset = torchvision.datasets.CIFAR10(root='./data',download=True)
Downloading https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz to ./data/cifar-10-python.tar.gz 
HBox(children=(FloatProgress(value=1.0, bar_style='info', max=1.0), HTML(value='')))
Extracting ./data/cifar-10-python.tar.gz to ./data 
test_set = torchvision.datasets.CIFAR10(root='./data',train=False)
train_set = torchvision.datasets.CIFAR10(root='./data',train=True)