Make TF CTRL compliant with XLA and AMP (#10209)

* Fix XLA and AMP

* Apply style

* Remove useless cast
This commit is contained in:
Julien Plu
2021-02-17 18:54:15 +01:00
committed by GitHub
parent fdb2351ebb
commit 7246785a67
2 changed files with 18 additions and 33 deletions

View File

@@ -222,14 +222,6 @@ class TFCTRLModelTest(TFModelTesterMixin, unittest.TestCase):
name = model.get_bias()
assert name is None
def test_mixed_precision(self):
# TODO JP: Make CTRL float16 compliant
pass
def test_xla_mode(self):
# TODO JP: Make CTRL XLA compliant
pass
@slow
def test_model_from_pretrained(self):
for model_name in TF_CTRL_PRETRAINED_MODEL_ARCHIVE_LIST[:1]: