阅读(4713) (0)

PyTorch深度学习

2020-09-05 14:03:00 更新

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

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

作者:  Soumith Chintala

本教程的目标:

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

本教程预先您对 numpy 有基本的了解。

注意

确保已安装 torch 和 torchvision 两者。

../_images/tensor_illustration_flat.png

什么是PyTorch?

../_images/autodiff.png

Autograd:自动分化

../_images/mnist1.png

神经网络

../_images/cifar101.png

训练分类器

../_images/data_parallel.png

可选:数据并行