Learn practical skills, build real-world projects, and advance your career
Updated 3 years ago
# Imports
import torch
import torchvision
from torchvision.datasets import MNIST
torchvision used for handle image data in PyTorch
# Download training dataset
dataset = MNIST(root='data/', download=True)C:\Users\OHIDUL ISLAM\.conda\envs\img2txt\lib\site-packages\torchvision\datasets\mnist.py:498: UserWarning: The given NumPy array is not writeable, and PyTorch does not support non-writeable tensors. This means you can write to the underlying (supposedly non-writeable) NumPy array using the tensor. You may want to copy the array to protect its data or make it writeable before converting it to a tensor. This type of warning will be suppressed for the rest of this program. (Triggered internally at ..\torch\csrc\utils\tensor_numpy.cpp:180.)
return torch.from_numpy(parsed.astype(m[2], copy=False)).view(*s)
len(dataset)60000We can create the test dataset using the MNIST class by passing train=False to the constructor.