examples/flax/language-modeling/run_mlm_flax.py · 7d321b7689c6ddc4d33ff93693a9e6ec2c65c51e · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[Flax] Allow retraining from save checkpoint (#12559) · 7d321b76
Patrick von Platen authored Jul 07, 2021
```
* fix_torch_device_generate_test

* remove @

* finish
```
7d321b76

run_mlm_flax.py 29 KB