趋近智
requires_grad)backward()).grad)torch.nn 搭建模型torch.nn.Module 基类torch.nn 损失)torch.optim)torch.utils.data.Datasettorchvision.transforms)torch.utils.data.DataLoader你已经学习了如何使用 torch.nn 定义模型,以及如何高效地使用 Dataset 和 DataLoader 准备数据。本章将侧重于整合这些组成部分,以实际训练你的神经网络。我们将构建被称为训练循环的标准迭代过程。
你将学习涉及的重要步骤:
DataLoader 提供的数据批次。$loss.backward() 计算损失相对于模型参数 (∇θL) 的梯度。$optimizer.step()。$optimizer.zero_grad() 将梯度归零。此外,我们将介绍如何实现一个独立的循环来评估模型在验证集或测试集上的表现,以及在训练期间或之后保存和加载模型状态(检查点)的方法。到本章结束时,你将能够为你的 PyTorch 模型实现一个完整的训练和评估流程。
6.1 训练循环的构成
6.2 设置模型、损失函数和优化器
6.3 使用 DataLoader 遍历数据
6.4 前向传播:获取预测结果
6.5 计算损失
6.6 反向传播:计算梯度
6.7 使用优化器更新权重
6.8 梯度清零
6.9 实现评估循环
6.10 保存和加载模型检查点
6.11 动手实践:完整训练流程
© 2026 ApX Machine Learning用心打造