transformers/tests/models/cvt
Gerald Cuder 5a2b77a6c1
Fix error in mixed precision training of TFCvtModel (#22267)
* Make sure CVT can be trained using mixed precision

* Add test for keras-fit with mixed-precision

* Update tests/models/cvt/test_modeling_tf_cvt.py

Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>

---------

Co-authored-by: gcuder <Gerald.Cuder@iacapps.com>
Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
2023-03-21 12:12:57 +00:00
..
__init__.py Add CvT (#17299) 2022-05-18 17:47:18 +02:00
test_modeling_cvt.py 🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516) 2023-02-28 19:40:57 +01:00
test_modeling_tf_cvt.py Fix error in mixed precision training of TFCvtModel (#22267) 2023-03-21 12:12:57 +00:00