transformers
Fix error in mixed precision training of `TFCvtModel`
#22267

Merged

Fix error in mixed precision training of `TFCvtModel` #22267

amyeroberts merged 3 commits into huggingface:main from gcuder:fix-cvt-mixed-precision-training

Make sure CVT can be trained using mixed precision

0e3d3909

Add test for keras-fit with mixed-precision

ca7dd1be

gcuder marked this pull request as ready for review 2 years ago

amyeroberts approved these changes on 2023-03-20

amyeroberts requested a review from

Rocketknight1 2 years ago

Rocketknight1 approved these changes on 2023-03-20

Update tests/models/cvt/test_modeling_tf_cvt.py

53cc1796

amyeroberts merged 5a2b77a6 into main 2 years ago

gcuder deleted the fix-cvt-mixed-precision-training branch 2 years ago

Reviewers

Rocketknight1

amyeroberts

Assignees

No one assigned

Labels

None yet

Milestone

No milestone