examples/language-modeling/run_mlm_flax.py · 2a7e8e1608aae8b5719a33c341401756e6e8897c · chenpangpang / transformers

Teven authored Dec 15, 2020

* replaced jnp.split + removing textual model inputs + ensuring warmup_steps > 0

* Add automatic dataset splitting in language-modeling examples

2a7e8e16

run_mlm_flax.py 27.2 KB