Add a test for mixed precision (#9806)

This commit is contained in:
Julien Plu
2021-01-27 09:36:49 +01:00
committed by GitHub
parent d5b40d6693
commit 2c891c156d
20 changed files with 105 additions and 8 deletions

View File

@@ -309,6 +309,10 @@ 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]: