优化模型参数

现在我们已经有了一个模型和数据，是时候通过优化模型参数来训练、验证和测试模型了。训练模型是一个迭代的过程：在每次迭代中，模型会对输出进行预测，计算预测的误差（损失），收集误差相对于参数的导数（在上一节中我们已看到），然后使用梯度下降优化这些参数。如果需要更详细的讲解，可以参考3Blue1Brown 的反向传播视频。

---

前置代码

我们加载了之前章节中的Datasets & DataLoaders和Build Model相关代码。

import torch
from torch