开始使用免费开始使用

手动更新权重

现在,您已经知道如何访问权重和偏置,接下来将手动完成 PyTorch 优化器的工作。虽然 PyTorch 会自动完成这些步骤,但手动练习有助于您直观理解模型如何学习和调整。在调试或微调神经网络时,这种理解非常有价值。

已经创建了一个包含 3 层的神经网络,并存储在变量 model 中。该网络已完成一次前向传播,并计算了损失及其导数。已经选择了一个默认学习率 lr,用于在更新时对梯度进行缩放。

本练习是课程的一部分

使用 PyTorch 的深度学习入门

查看课程

交互式实操练习

通过完成这段示例代码来试试这个练习。

weight0 = model[0].weight
weight1 = model[1].weight
weight2 = model[2].weight

# Access the gradients of the weight of each linear layer
grads0 = ____
grads1 = ____
grads2 = ____
编辑并运行代码