# Imports
import torch
import torchvision 
from torchvision.datasets import MNIST

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

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

len(dataset)

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