"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "963db81a5afc5c31edafa0c9fb0390956a78cf2a"
Unverified Commit de5f86e5 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Skip `TFCvtModelTest::test_keras_fit_mixed_precision` for now (#23699)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 3d574044
......@@ -186,6 +186,7 @@ class TFCvtModelTest(TFModelTesterMixin, PipelineTesterMixin, unittest.TestCase)
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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment