tests/big_bird/test_modeling_big_bird.py · f571dc20ac27c3da2076acf5d53a5acaad6b6c5d · chenpangpang / transformers

Update PT Flax equivalence tests in PT test file (#16280) · f571dc20

Yih-Dar authored Mar 24, 2022



* update PT/Flax equivalence tests on PT side

* overwrite check_outputs in BigBirdModelTest
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

f571dc20

test_modeling_big_bird.py 41.6 KB

Replace test_modeling_big_bird.py