PyTorch 深度学习: 60 分钟极速入门

优质
小牛编辑
138浏览
2023-12-01

译者:bat67

作者:Soumith Chintala

此教程的目标:

  • 更高层次地理解Pythrch的Tensor库以及神经网络。
  • 训练一个小的神经网络模型用于分类图像。

本教程假设读者对numpy有基本的了解

确保你安装了 torch 和 torchvision 包。

https://pytorch.org/tutorials/_images/tensor_illustration_flat.png

PyTorch 是什么?

https://pytorch.org/tutorials/_images/autodiff.png

Autograd:自动求导

https://pytorch.org/tutorials/_images/mnist1.png

神经网络

https://pytorch.org/tutorials/_images/cifar101.png

训练分类器

https://pytorch.org/tutorials/_images/data_parallel.png

可选:数据并行