We will use the famous MNIST Handwritten Digits Databases as our training dataset.It consists of 28px by 28px grayscale images of handwritten disgits(0–9), along with labels for each image indicating which digit it represents. MNIST stands for Modified National Institute of Standards and Technology.PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.
import torchvision ## Contains some utilities for working with the image data
from torchvision.datasets import MNIST
import matplotlib.pyplot as plt
import torchvision.transforms as transforms
from torch.utils.data import random_split
from torch.utils.data import DataLoader
import torch.nn.functional as F
In the ever-changing ecosystem of convolution neural network (CNN), Recently I read an interesting article on “Alexnet Architecture”. I decided to unravel the learning of amazing paper on Alexnet.AlexNet is the name of a Convolution Neural Network, designed by Alex Krizhevskly,and published with Ilya Sutskever and Krizhevsky’s PhD advisor Geoffrey Hinton.
AlexNet competed in the ImageNet Large Scale Visual Recognition Challenge in 2012. The Network achieved a top-5 error of 15.3%.
Key Points of this Architecture ->
1. Non-Saturating Non-linearity: In this Architecture, we have used ReLu Activation function. …