optax.ScaleByZoomLinesearchState#
- class optax.ScaleByZoomLinesearchState(learning_rate: jax.typing.ArrayLike, value: jax.typing.ArrayLike, grad: base.Updates, info: ZoomLinesearchInfo)[source]#
State for scale_by_zoom_linesearch.
- learning_rate#
learning rate computed at the end of a round of line-search, used to scale the update.
- Type:
jax.typing.ArrayLike
- value#
value of the objective computed at the end of a round of line-search. Can be reused using
optax.value_and_grad_from_state().- Type:
jax.typing.ArrayLike
- grad#
gradient of the objective computed at the end of a round of line-search. Can be reused using
optax.value_and_grad_from_state().- Type:
base.Updates
- info#
Additional information on the status of the linesearch see
optax.ZoomLinesearchInfo.- Type: