Suraj Patil
|
85a4bda4f4
|
bump flax version (#14343)
|
2021-11-09 22:15:22 +05:30 |
|
Suraj Patil
|
2d42915abe
|
[examples/flax] add adafactor optimizer (#12544)
* add adafactor
* Update examples/flax/language-modeling/run_mlm_flax.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
|
2021-07-07 11:50:30 +05:30 |
|
Avital Oliver
|
9b393240a2
|
Use a released version of optax rather than installing from Git. (#12173)
Use a released version of optax rather than installing from Git
|
2021-06-15 16:42:51 +05:30 |
|
Patrick von Platen
|
00440e350f
|
[Flax MLM] Refactor run mlm with optax (#11745)
* refactor
* update
* update
* update
* refactor run mlm
* finalize
* refactor more
* fix typo
* update
* finish refactor
* modify run mlm
* Apply suggestions from code review
* Apply suggestions from code review
* Apply suggestions from code review
* small fixes
* upload
* upload
* finish run mlm script
Co-authored-by: Patrick von Platen <patrick@huggingface.co>
|
2021-05-19 12:00:58 +01:00 |
|