examples/flax/language-modeling/run_t5_mlm_flax.py · b90096fe1424e1927e549b73b5db67341a0d27d0 · chenpangpang / transformers

[Flax T5] Speed up t5 training (#13012) · 2e408236

Patrick von Platen authored Aug 06, 2021



* fix_torch_device_generate_test

* remove @

* update

* up

* fix

* remove f-stings

* correct readme

* up
Co-authored-by: Patrick von Platen <patrick@huggingface.co>

2e408236

run_t5_mlm_flax.py 34.7 KB

Replace run_t5_mlm_flax.py