Deep Learning with PyTorch: A 60 Minute Blitz¶
Author: Soumith Chintala
What is PyTorch?¶
PyTorch is a Python-based scientific computing package serving two broad purposes:
A replacement for NumPy to use the power of GPUs and other accelerators.
An automatic differentiation library that is useful to implement neural networks.
Goal of this tutorial:¶
Understand PyTorch’s Tensor library and neural networks at a high level.
Train a small neural network to classify images