Commits · bd9871657bb9500a9f4437a873db6df5f1ae6dbb · chenpangpang / transformers

21 May, 2021 1 commit

[Flax] Align GLUE training script with mlm training script (#11778) · bd987165

Patrick von Platen authored May 21, 2021



* speed up flax glue

* remove unnecessary line

* remove folder

* remove run in loop
Co-authored-by: Patrick von Platen <patrick@huggingface.co>

bd987165

17 May, 2021 1 commit

Improvements to Flax finetuning script (#11727) · 726e953d

Marc van Zee authored May 17, 2021

* Add Cloud details to README

* Flax script and readme updates

* Some simplifications of Flax script

726e953d

14 May, 2021 1 commit
- Add Cloud details to README (#11706) · 94a23487
  Marc van Zee authored May 14, 2021
```
* Add Cloud details to README

* Flax script and readme updates
```
  94a23487
12 May, 2021 1 commit
- Updates README and fixes bug (#11701) · 6797cdc0
  Marc van Zee authored May 12, 2021
  
  6797cdc0
11 May, 2021 1 commit

Adds Flax BERT finetuning example on GLUE (#11564) · 4ce6bcc3

Marc van Zee authored May 11, 2021



* Adds Flax BERT finetuning example

* fix traced jax tensor type

* Use Optax losses and learning schedulers

* Add 1GPU training results

* merge into master & make style

* fix input

* del file

* Fix bug in loss and add torch runs

* finish bert flax fine-tune

* Update examples/flax/text-classification/README.md

* Update examples/flax/text-classification/run_flax_glue.py

* add requirements

* finalize

* finalize
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Patrick von Platen <patrick@huggingface.co>

4ce6bcc3