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

Co-authored-by: Masahiro Masuda <masahi@[email protected]>
jroesch · Jun 29, 2021 · 61a6ea1 · 61a6ea1
1 parent 8cc754c
commit 61a6ea1
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
@@ -3469,7 +3469,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)