Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
4 years ago
import numpy as np
import torch
import torchvision
from torchvision.datasets import MNIST
from torchvision.transforms import ToTensor
from torch.utils.data.sampler import SubsetRandomSampler
from torch.utils.data.dataloader import DataLoader

Dateset prepearing

dataset = MNIST(root='data/',
               download=True,
               transform=ToTensor())
dataset
Dataset MNIST
    Number of datapoints: 60000
    Root location: data/
    Split: Train
    StandardTransform
Transform: ToTensor()
img, label = dataset[0]
img.shape, label
(torch.Size([1, 28, 28]), 5)