Add mathematical formula in docstring of linear_schedule

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 copybara-service[bot] closed 6 months ago

copybara-service[bot] commented 6 months ago

Add mathematical formula in docstring of linear_schedule

For consistency I also changed notation \gamma_0 -> I in cosine_decay_schedule.