Rate this Page
★★★★★

torch.optim.Optimizer.step#

Optimizer.step(closure:None=None)→None[source]#
Optimizer.step(closure:Callable[[],float])→float

Perform a single optimization step to update parameter.

Parameters

closure (Callable) – A closure that reevaluates the model andreturns the loss. Optional for most optimizers.