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_blenderbot_small.py
+++ b/tests/test_modeling_tf_blenderbot_small.py
@@ -279,6 +279,10 @@ class TFBlenderbotSmallModelTest(TFModelTesterMixin, unittest.TestCase):
        # This test is too long (>30sec) and makes fail the CI
        pass

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

 def _assert_tensors_equal(a, b, atol=1e-12, prefix=""):
    """If tensors not close, or a and b arent both tensors, raise a nice Assertion error."""