SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add a test for mixed precision (#9806)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_modeling_tf_funnel.py
View File

@@ -371,6 +371,10 @@ class TFFunnelModelTest(TFModelTesterMixin, unittest.TestCase):
# This test is too long (>30sec) and makes fail the CI
pass
def test_mixed_precision(self):
# TODO JP: Make Funnel float16 compliant
pass
@require_tf
class TFFunnelBaseModelTest(TFModelTesterMixin, unittest.TestCase):
@@ -401,3 +405,7 @@ class TFFunnelBaseModelTest(TFModelTesterMixin, unittest.TestCase):
def test_saved_model_creation(self):
# This test is too long (>30sec) and makes fail the CI
pass
def test_mixed_precision(self):
# TODO JP: Make Funnel float16 compliant
pass