Pytorch Mnist - Notebook by ahmadyahya11 (ahmadyahya11)

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

Updated 3 years ago

# Imports
import jovian
import torch
import torchvision
from torchvision.datasets import MNIST
import torch.nn.functional as F 
from torch.utils.data import random_split
from torch.utils.data import DataLoader
import torch.nn as nn

Loading the dataset

# Download training datasets
dataset = MNIST(root='data/', download=True)

len(dataset)

test_dataset = MNIST(root='data/', train=False)
len(test_dataset)