search

google / jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.
https://jaxopt.github.io
Apache License 2.0
939 stars 66 forks source link

AndersonAcceleration mismatch between code and documentation #443

Closed sophie-xhonneux closed 1 year ago

sophie-xhonneux commented 1 year ago

The documentation at https://jaxopt.github.io/stable/_autosummary/jaxopt.AndersonAcceleration.html#jaxopt.AndersonAcceleration states in the detailed description of parameters that is has an init value has_aux but doesn't in the code and naturally fails when one tries to set has_aux at initialisation.

Would it be possible to add the has_aux flag to the AndersonAcceleration?

Algue-Rythme commented 1 year ago

Hi Sophie Nice catch, we will solve that!