Fixed the Sharpness aware minimization docstring

google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.

https://optax.readthedocs.io

Apache License 2.0

1.63k stars 174 forks source link

Closed hmludwig closed 6 months ago

hmludwig commented 6 months ago

Fixed the docstring such that it gets displayed correctly (see #826)

hmludwig commented 6 months ago

Thanks for the suggestions. I adopted them :)

fabianp commented 6 months ago

amazing, let's get this merged! 🚀

hmludwig commented 6 months ago

I accidentally pushed to the wrong branch. It seems my reset messed things up. Sorry for that.

fabianp commented 6 months ago

no worries, we can easily get the original code