Skip TFCvtModelTest::test_keras_fit_mixed_precision for now (huggin…

…gface#23699) fix Co-authored-by: ydshieh <[email protected]>
gojiteji · Jun 5, 2023 · ee663b3 · ee663b3
1 parent 51da3be
commit ee663b3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)