Expanded Library

Building Deep Learning Models Using PyTorch

by Janani Ravi

PyTorch is an open source deep learning framework originally developed by the AI teams at Facebook. PyTorch offers high-level APIs which make it easy to build neural networks and great support for distributed training and prediction.

What you'll learn

PyTorch is an open source, deep learning framework which is a popular alternative to TensorFlow and Apache MXNet. PyTorch APIs follow a Python-native approach which, along with dynamic graph execution, make it very intuitive to work with for Python developers and data scientists.

In this course, Building Deep Learning Models Using PyTorch, you will learn to work with PyTorch and all the libraries that it has to offer, from first principles - starting with Torch tensors, dynamic computation graphs, and the autograd library, to compute gradients.

You'll start off by understanding the basics of training a neural network, the forward and backward passes, and gradient computation. You will use these concepts to build simple neural networks to predict automobile prices, as well as who survived and who did not on the Titanic.

Next, you'll move on to image classification using convolutional neural networks; you'll study the role of convolutional and pooling layers and the basic structure of a CNN, you'll then build a CNN to classify images from the Cifar-10 dataset. You'll also see how you can leverage the power of transfer learning by using pre-trained models for image classification.

Finally, you'll get to work with recurrent neural networks for sequence data, seeing how the dynamic computation graph execution in PyTorch makes building RNNs very simple. You'll use RNNs with long memory cells to predict gender using baby names.

At the end of this course, you will be comfortable using PyTorch libraries and APIs to leverage pre-trained models that PyTorch offers and also to build your own custom model for your specific use case.

Table of contents

Course Overview

About the author

Janani has a Masters degree from Stanford and worked for 7+ years at Google. She was one of the original engineers on Google Docs and holds 4 patents for its real-time collaborative editing framework. After spending years working in tech in the Bay Area, New York, and Singapore at companies such as Microsoft, Google, and Flipkart, Janani finally decided to combine her love for technology with her passion for teaching. She is now the co-founder of Loonycorn, a content studio focused on providin... more

Ready to upskill? Get started