使用 PyTorch 进行深度学习:60 分钟的闪电战

原文: https://pytorch.org/tutorials/beginner/deep_learning_60min_blitz.html

作者Soumith Chintala

本教程的目标:

  • 全面了解 PyTorch 的 Tensor 库和神经网络。
  • 训练一个小型神经网络对图像进行分类

本教程假定您对 numpy 有基本的了解。

注意

确保已安装torchtorchvision软件包。

../_images/tensor_illustration_flat.png

什么是 PyTorch?

../_images/autodiff.png

Autograd:自动分化

../_images/mnist1.png

神经网络

../_images/cifar101.png

训练分类器

../_images/data_parallel.png

可选:数据并行