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

FIX: broken example resnet_flax #352

Closed fabianp closed 1 year ago

fabianp commented 1 year ago

The call to init_state was raising an exception. Not sure if the API has recently change or whether it was always been broken ...

Algue-Rythme commented 1 year ago

It has always been broken I think, even on the .py file: see #328 The reason was that aux was not computed. Once this bug was corrected it triggered this one.

fabianp commented 1 year ago

Thanks @Algue-Rythme ! do you think this fix is correct?

Algue-Rythme commented 1 year ago

Yes I guess so!