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_modeling_data2vec_audio.py

[Data2Vec] Speed up test (#17660)

2022-06-10 18:48:58 +02:00

test_modeling_data2vec_text.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00

test_modeling_data2vec_vision.py

Improve vision models (#17731)

2022-06-24 11:34:51 +02:00

test_modeling_tf_data2vec_vision.py

Add TFData2VecVision for semantic segmentation (#17271)

2022-06-08 14:03:18 +01:00