Home AI | Tech | Machine Learning | Data Science

AI | Tech | Machine Learning | Data Science


Introduction to Image Classification using Pytorch to Classify FashionMNIST Dataset

In this blog post, we will discuss how to build a Convolution Neural Network that can classify Fashion MNIST data using Pytorch on Google Colaboratory. The way we do that is, first we will download the data using Pytorch DataLoader class and then we will use LeNet-5 architecture to build our model. Finally, we will train our model on GPU and evaluate it on the test data.

Artificial Intelligence

Data Science

Tech News

Join Our 16k+ AI/ Machine Learning/ Deep Learning/ Data Science Facebook Group

Join Now