optax icon indicating copy to clipboard operation
optax copied to clipboard

Add an example of reading the learning rate from the optimizer state

Open rosshemsley opened this issue 3 years ago • 2 comments

This question comes up relatively often, let's put it clearly in the documentation.

rosshemsley avatar Mar 04 '22 08:03 rosshemsley

I'd like to work on this!

yulonglin avatar Mar 08 '22 23:03 yulonglin

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!

mkunesch avatar Mar 18 '22 17:03 mkunesch