Deep Learning with PyTorch: A 60 Minute Blitz¶

Author: Soumith Chintala

Goal of this tutorial:

Understand PyTorch’s Tensor library and neural networks at a high level.
Train a small neural network to classify images

This tutorial assumes that you have a basic familiarity of numpy

Note

Make sure you have the torch and torchvision packages installed.

../_images/tensor_illustration_flat.png

What is PyTorch?

../_images/Variable1.png

Autograd: automatic differentiation

../_images/mnist1.png

Neural Networks

../_images/cifar101.png

Training a classifier