tests/test_modeling_flax_common.py · 604c0850878c1840e19e025f6f909c5956716aa6 · chenpangpang / transformers

"examples/research_projects/adversarial/run_hans.py" did not exist on "89d627289890c8e556df72b7901260b038236e55"

[Flax] Adapt Flax models to new structure (#9484) · 0b98ca36

Patrick von Platen authored Mar 18, 2021



* Create modeling_flax_eletra with code copied from modeling_flax_bert

* Add ElectraForMaskedLM and ElectraForPretraining

* Add modeling test for Flax electra and fix naming and arg in Flax Electra model

* Add documentation

* Fix code style

* Create modeling_flax_eletra with code copied from modeling_flax_bert

* Add ElectraForMaskedLM and ElectraForPretraining

* Add modeling test for Flax electra and fix naming and arg in Flax Electra model

* Add documentation

* Fix code style

* Fix code quality

* Adjust tol in assert_almost_equal due to very small difference between model output, ranging 0.0010 - 0.0016

* Remove redundant ElectraPooler

* save intermediate

* adapt

* correct bert flax design

* adapt roberta as well

* finish roberta flax

* finish

* apply suggestions

* apply suggestions
Co-authored-by: Chris Nguyen <anhtu2687@gmail.com>

0b98ca36

test_modeling_flax_common.py 6 KB

Replace test_modeling_flax_common.py