I was trying to use jaxopt.LBFGS with linesearch='hager-zhang'.
The following error appeared
Attempted boolean conversion of traced array with shape bool[]..
The error occurred while tracing the function _update_interval at python3.10/site-packages/jaxopt/_src/hager_zhang_linesearch.py:286 for cond. This value became a tracer due to JAX operations on these lines:
operation a:f64[] = convert_element_type[new_dtype=float64 weak_type=False] b
This error occurs when trying to make the update using the .update() methods.
I was trying to use
jaxopt.LBFGS
withlinesearch='hager-zhang'
. The following error appeared.update()
methods.zoom
orbacktracking