optax
optax copied to clipboard
Add an example of reading the learning rate from the optimizer state
This question comes up relatively often, let's put it clearly in the documentation.
I'd like to work on this!
That's great, thanks a lot! The replies in #206 might be helpful for showing the different ways of doing it. Let us know if you have any questions!