Building neural network with PyTorch

Previously we were talking about building supervised learning models from scratch. And now it’s time to roll out big guns. I’m going to use the PyTorch library to build a model which is able to recognize handwritten digits. After writing everything from scratch It’s an amazing feeling of how PyTorch eases the process of developing and testing deep learning models. Let’s take a look at what I got.

Image by Soumith Chintala