1.2 构建模型

在1.1节中，我们对MNIST数据集有了基本的认识，并掌握了如何在PyTorch中导入该数据集，本节将以PyTorch为工具，搭建一个图像识别网络。一个典型的神经网络训练过程包括定义神经网络、前向传播、计算损失、反向传播、更新参数。