Image Classification using Pre-trained Models in PyTorch

This post is part of the series in which we are going to cover the following topics. In the previous blog we discussed about PyTorch, it's strengths and why should you learn it. We also had a brief look at Tensors – the core data structure in PyTorch. In this blog, we will jump into some hands-on examples of using pre-trained networks present in TorchVision module for Image Classification. Torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.