tests/vit/test_modeling_flax_vit.py · d3c9d0e55fad0096e7b07fd9be3c411ecd34135c · chenpangpang / transformers

[ViT, BEiT, DeiT, DPT] Improve code (#16799) · d3c9d0e5

NielsRogge authored Apr 18, 2022



* Improve code

* Fix bugs

* Fix another bug

* Clean up DTP as well

* Update DPT model outputs
Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>

d3c9d0e5

test_modeling_flax_vit.py 9.68 KB

Replace test_modeling_flax_vit.py