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)
60000
test_dataset = MNIST(root='data/', train=False)
len(test_dataset)
10000