Making TF MobileBert model compliant with AMP (#10259)

* Fix AMP * Trigger CI * Rework cast
2021-02-19 12:55:25 +01:00
parent 2fc6284f04
commit fb56bf2584
2 changed files with 15 additions and 16 deletions
--- a/tests/test_modeling_tf_mobilebert.py
+++ b/tests/test_modeling_tf_mobilebert.py
@@ -310,10 +310,6 @@ class TFMobileBertModelTest(TFModelTesterMixin, unittest.TestCase):
        # This test is too long (>30sec) and makes fail the CI
        pass

-    def test_mixed_precision(self):
-        # TODO JP: Make MobileBert float16 compliant
-        pass
-
    @slow
    def test_model_from_pretrained(self):
        # for model_name in TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]: