transformers
d481b641 - Make Flax pt-flax equivalence test more aggressive (#15841)
Commit
View On GitHub
Commit
3 years ago
Make Flax pt-flax equivalence test more aggressive (#15841) * Make test_equivalence_pt_to_flax more aggressive * Make test_equivalence_flax_to_pt more aggressive * don't use to_tuple * clean-up * fix missing test cases + testing on GPU * fix conversion * fix `ValueError: assignment destination is read-only` * Add type checking * commit to revert later * Fix * fix * fix device * better naming * clean-up Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
Author
ydshieh ydshieh
Parents
c03b6e42
FAQ Terms Privacy Refunds Impressum
Loading