oneflow.optim.Optimizer.step¶
-
Optimizer.
step
(closure: Optional[Callable] = None) → Optional[oneflow.Tensor]¶ Performs a single optimization step (parameter update).
- Parameters
closure (Union[Callable, None], optional) – A closure that reevaluates the model and returns the loss. Optional for most optimizers.
- Returns
The loss.
- Return type
Union[Tensor, None]