examples/flax/language-modeling/run_mlm_flax.py · 13a9c9a354d07f7c21e968f2ac4c7c948b7307a0 · chenpangpang / transformers

[Flax] Refactor gpt2 & bert example docs (#13024) · 13a9c9a3

Patrick von Platen authored Aug 09, 2021



* fix_torch_device_generate_test

* remove @

* improve docs for clm

* speed-ups

* correct t5 example as well

* push final touches

* Update examples/flax/language-modeling/README.md

* correct docs for mlm

* Update examples/flax/language-modeling/README.md
Co-authored-by: Patrick von Platen <patrick@huggingface.co>

13a9c9a3

run_mlm_flax.py 29.5 KB

Replace run_mlm_flax.py