Making TF Lxmert model compliant with AMP (#10257)

* Fix AMP * Rework cast * Apply style
2021-02-19 12:54:14 +01:00
parent d27b28d958
commit 2fc6284f04
2 changed files with 16 additions and 15 deletions
--- a/tests/test_modeling_tf_lxmert.py
+++ b/tests/test_modeling_tf_lxmert.py
@@ -706,10 +706,6 @@ class TFLxmertModelTest(TFModelTesterMixin, unittest.TestCase):
        # This test is too long (>30sec) and makes fail the CI
        pass

-    def test_mixed_precision(self):
-        # TODO JP: Make Lxmert float16 compliant
-        pass
-
    @slow
    def test_saved_model_creation_extended(self):
        config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()