[TEST] Disable flaky TF combined NMS test (apache#8364)

Co-authored-by: Masahiro Masuda <masahi@129@gmail.com>
zxy844288792 · Jan 13, 2022 · 04ed0be · 04ed0be
1 parent f00be1f
commit 04ed0be
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py
@@ -3484,7 +3484,9 @@ def test_forward_combined_nms():
     _test_forward_combined_nms((1, 32, 1, 4), (1, 32, 1), 0.7, 0.5, 10, 64)
     _test_forward_combined_nms((1, 32, 1, 4), (1, 32, 2), 0.7, 0.5, 32, 64)
     _test_forward_combined_nms((1, 64, 1, 4), (1, 64, 20), 0.7, 0.5, 64, 10)
-    _test_forward_combined_nms((1, 64, 20, 4), (1, 64, 20), 0.7, 0.5, 64, 64, clip_boxes=True)
+    # This workload seems flaky on CI.
+    # See https://github.com/apache/tvm/issues/8140
+    # _test_forward_combined_nms((1, 64, 20, 4), (1, 64, 20), 0.7, 0.5, 64, 64, clip_boxes=True)
     _test_forward_combined_nms((2, 200, 1, 4), (2, 200, 1), 0.4, 0.6, 100, 100)
     _test_forward_combined_nms((2, 200, 1, 4), (2, 200, 10), 0.4, 0.2, 150, 1000)