Making TF TransfoXL model compliant with AMP (#10264)

* Fix AMP * Apply style * Remove unused import
2021-02-19 12:58:07 +01:00
parent 86caeb7636
commit 3e116ed331
3 changed files with 41 additions and 24 deletions
--- a/tests/test_modeling_tf_transfo_xl.py
+++ b/tests/test_modeling_tf_transfo_xl.py
@@ -205,10 +205,6 @@ class TFTransfoXLModelTest(TFModelTesterMixin, unittest.TestCase):
                name = model.get_bias()
                assert name is None

-    def test_mixed_precision(self):
-        # TODO JP: Make TransfoXL float16 compliant
-        pass
-
    def test_xla_mode(self):
        # TODO JP: Make TransfoXL XLA compliant
        pass