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]