From ee663b37a00f3fedfc2a54cc82f355a161ee083e Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 23 May 2023 20:47:47 +0200 Subject: [PATCH] Skip `TFCvtModelTest::test_keras_fit_mixed_precision` for now (#23699) fix Co-authored-by: ydshieh --- tests/models/cvt/test_modeling_tf_cvt.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/cvt/test_modeling_tf_cvt.py b/tests/models/cvt/test_modeling_tf_cvt.py index 484bd295d17291..f0f7b9c5d823ba 100644 --- a/tests/models/cvt/test_modeling_tf_cvt.py +++ b/tests/models/cvt/test_modeling_tf_cvt.py @@ -186,6 +186,7 @@ def test_dataset_conversion(self): def test_keras_fit(self): super().test_keras_fit() + @unittest.skip(reason="Get `Failed to determine best cudnn convolution algo.` error after using TF 2.12+cuda 11.8") def test_keras_fit_mixed_precision(self): policy = tf.keras.mixed_precision.Policy("mixed_float16") tf.keras.mixed_precision.set_global_policy(policy)