transformers/tests/models/cvt
Mathieu Jouffroy 5ca131f3d4
[CvT] Tensorflow implementation (#18597)
* implemented TFCvtModel and TFCvtForImageClassification and modified relevant files, added an exception in convert_tf_weight_name_to_pt_weight_name, added quick testing file to compare with pytorch model

* added docstring + testing file in transformers testing suite

* added test in testing file, modified docs to pass repo-consistency, passed formatting test

* refactoring + passing all test

* small refacto, removing unwanted comments

* improved testing config

* corrected import error

* modified acces to pretrained model archive list, to pass tf_test

* corrected import structure in init files

* modified testing for keras_fit with cpu

* correcting PR issues + Refactoring

* Refactoring : improving readability and reducing the number of permutations

* corrected momentum value + cls_token initialization

* removed from_pt as weights were added to the hub

* Update tests/models/cvt/test_modeling_tf_cvt.py

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
2022-10-11 18:16:52 +01:00
..
__init__.py Add CvT (#17299) 2022-05-18 17:47:18 +02:00
test_modeling_cvt.py has_attentions - consistent test skipping logic and tf tests (#17495) 2022-06-09 09:50:03 +02:00
test_modeling_tf_cvt.py [CvT] Tensorflow implementation (#18597) 2022-10-11 18:16:52 +01:00