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

第 3 节介绍了训练一个神经网络的常见步骤,包括定义具有一些可学习参数的神经网络、遍历输入数据集、通过神经网络处理输入、计算损失、将梯度传播回网络参数、更新网络权重等。

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

图4/7

在学会定义网络、计算损失、更新权重之后,第 4 节开始讲解如何训练一个分类器。教程使用了 CIFAR10 数据集,将训练步骤分为 5 步:

1. 载入 CIFAR10 并将其标准化;

2. 定义一个卷积神经网络;

3. 定义损失函数和优化器;

4. 训练网络;

5. 在测试集上测试网络

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

图5/7

推荐阅读