60分钟入门PyTorch,官方教程手把手教你训练第一个深度学习模型( 四 )

CIFAR10 数据集。

此外,这一节还讲解了如何在 GPU 上训练神经网络。

如果想进一步加速训练过程,还可以选修第 5 节——数据并行,学习如何在多个 GPU 上训练网络。

在这一教程中,每个小节都有 GoogleColab 链接,可以让学习者实时运行代码,获取实战经验。

60分钟入门PyTorch,官方教程手把手教你训练第一个深度学习模型

图6/7

如果想在本地运行文件,还可以下载 Notebook。

参考链接:

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

https://news.ycombinator.com/item?id=21240057

https://www.youtube.com/watch?v=u7x8RXwLKcA

机器之心「SOTA模型」:22大领域、127个任务,机器学习 SOTA 研究一网打尽。

推荐阅读