Add a test for mixed precision (#9806)

2021-01-27 09:36:49 +01:00
parent d5b40d6693
commit 2c891c156d
20 changed files with 105 additions and 8 deletions
--- a/tests/test_modeling_tf_t5.py
+++ b/tests/test_modeling_tf_t5.py
@@ -306,6 +306,10 @@ class TFT5ModelTest(TFModelTesterMixin, unittest.TestCase):
        # This test is too long (>30sec) and makes fail the CI
        pass

+    def test_mixed_precision(self):
+        # TODO JP: Make T5 float16 compliant
+        pass
+
    @slow
    def test_model_from_pretrained(self):
        model = TFT5Model.from_pretrained("t5-small")
@@ -435,6 +439,10 @@ class TFT5EncoderOnlyModelTest(TFModelTesterMixin, unittest.TestCase):
    def test_train_pipeline_custom_model(self):
        pass

+    def test_mixed_precision(self):
+        # TODO JP: Make T5 float16 compliant
+        pass
+

@require_tf
@require_sentencepiece