[pytoch文档代码注释]2.构建CNN、损失函数和优化器

训练图像分类器

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

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

  1. 定义卷积神经网络

  2. 定义损失函数

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

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

关于卷积和池化输出tensor的shape(size)计算,可以查看 b站up 和 的视频讲解。————-

           

资源下载: