The gradient
A line
The initial step size
Once learning is done, the weights should be copied back into normal tensors.
Once learning is done, the weights should be copied back into normal tensors.
The weights
The gradient
The initial step size
Some optimizers swap out weights with special purpose tensors for e.
Some optimizers swap out weights with special purpose tensors for e.g. efficient scoring while learning.
The weights
Whether the optimizer has converged yet.
Whether the optimizer has converged yet.
A line
Reset the optimizers internal state (such as Hessian approximation, etc.
Reset the optimizers internal state (such as Hessian approximation, etc.)
Updates the weights according to the gradient.
Updates the weights according to the gradient.
The weights
The gradient
The value
A backtracking line optimizer. Shouldn't be used directly.