riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-29 09:12:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity
3a1a56a8fe
transformers/tests/models/vit
History
NielsRogge 0917870510
Improve vision models (#17731) 
* Improve vision models

* Add a lot of improvements

* Remove to_2tuple from swin tests

* Fix TF Swin

* Fix more tests

* Fix copies

* Improve more models

* Fix ViTMAE test

* Add channel check for TF models

* Add proper channel check for TF models

* Apply suggestion from code review

* Apply suggestions from code review

* Add channel check for Flax models, apply suggestion

* Fix bug

* Add tests for greyscale images

* Add test for interpolation of pos encodigns

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-06-24 11:34:51 +02:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_feature_extraction_vit.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_vit.py Improve vision models (#17731) 2022-06-24 11:34:51 +02:00
test_modeling_tf_vit.py Improve vision models (#17731) 2022-06-24 11:34:51 +02:00
test_modeling_vit.py Improve vision models (#17731) 2022-06-24 11:34:51 +02:00