[pytoch文件代码注释]3.培训和；储蓄及；加载和加载；测试CNN网络

训练图像分类器

我们将按顺序执行以下步骤：

我们将按顺序执行以下步骤：

4. 根据训练数据训练网络

5. 在测试数据上测试网络

我们如何在 GPU 上运行这些神经网络？

在 GPU 上进行训练

就像将张量转移到 GPU 上一样，您也将神经网络转移到 GPU 上。

如果可以使用 CUDA，首先将我们的设备定义为第一个可见的 cuda 设备：

本节的其余部分假定device是 CUDA 设备。

然后，这些方法将递归遍历所有模块，并将其参数和缓冲区转换为 CUDA 张量：

请记住，您还必须将每一步的输入和目标也发送到 GPU：