torch.optim.Optimizer.step

优化器.step(closure:None=None) None[源代码]
Optimizer.step(closure:Callable[[],float]) float

执行一个优化步骤来更新参数。

参数

closure (Callable) – 一个可选的功能,用于重新评估模型并返回损失值。

注意

除非另有说明,否则此函数不应修改参数的.grad字段。

本页目录