torch.optim.Optimizer.step — PyTorch 2.0 documentation (original) (raw)

Optimizer.step(closure)[source]¶

Performs a single optimization step (parameter update).

Parameters:

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

Note

Unless otherwise specified, this function should not modify the.grad field of the parameters.

• torch.optim.Optimizer.step

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.