tests/test_modeling_flax_common.py · f47afefb210a120545a5e9745292aaf6c316d246 · chenpangpang / transformers

Use 5e-5 For BigBird PT/Flax equivalence tests (#17780) · f47afefb

Yih-Dar authored Jun 21, 2022



* rename to check_pt_flax_outputs

* update check_pt_flax_outputs

* use 5e-5 for BigBird PT/Flax test
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

f47afefb

test_modeling_flax_common.py 48.7 KB

Replace test_modeling_flax_common.py