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

Fix tests of mixed precision now that experimental is deprecated (#17300)

Browse Source 
* Fix tests of mixed precision now that experimental is deprecated

* Fix mixed precision in training_args_tf.py too
This commit is contained in:
Matt
2022-05-17 14:14:17 +01:00
committed by GitHub
parent 6d211429ec
commit 651e48e1e5
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/utils/test_modeling_tf_core.py
View File

@@ -205,7 +205,7 @@ class TFCoreModelTesterMixin:
@slow
def test_mixed_precision(self):
tf.keras.mixed_precision.experimental.set_policy("mixed_float16")
tf.keras.mixed_precision.set_global_policy("mixed_float16")
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
@@ -216,7 +216,7 @@ class TFCoreModelTesterMixin:
self.assertIsNotNone(outputs)
tf.keras.mixed_precision.experimental.set_policy("float32")
tf.keras.mixed_precision.set_global_policy("float32")
@slow
def test_train_pipeline_custom_model(self):