flax
8ac99b8f - Merge pull request #352 from wrzadkow:assert-lamb

Commit

5 years ago

Merge pull request #352 from wrzadkow:assert-lamb PiperOrigin-RevId: 321593370

References

#352 - Assert valid learning rate in LAMB optimizer

Author

a-googler

a-googler

Parents

Loading