tests/models/cvt/test_modeling_tf_cvt.py · 3d57404464516cbddf22d6ae43dea59c233fe53f · chenpangpang / transformers

Fix error in mixed precision training of `TFCvtModel` (#22267) · 5a2b77a6

Gerald Cuder authored Mar 21, 2023



* 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>

5a2b77a6

test_modeling_tf_cvt.py 10.5 KB

Replace test_modeling_tf_cvt.py