Skip to content

Commit

Permalink
[CI][TEST] Temporary disable nmsv4 test (#6151)
Browse files Browse the repository at this point in the history
  • Loading branch information
tqchen authored Jul 28, 2020
1 parent a43d69d commit 5a31251
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/python/frontend/tensorflow/test_forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -2051,7 +2051,7 @@ def _test_forward_nms_v4(bx_shape, score_shape, iou_threshold, score_threshold,

def test_forward_nms():
""" NonMaxSuppressionV3,4 """
for _test_forward_nms in [_test_forward_nms_v3, _test_forward_nms_v4]:
for _test_forward_nms in [_test_forward_nms_v3]:
_test_forward_nms((5, 4), (5,), 0.7, 0.5, 5)
_test_forward_nms((20, 4), (20,), 0.5, 0.6, 10)
_test_forward_nms((1000, 4), (1000,), 0.3, 0.7, 1000)
Expand Down

0 comments on commit 5a31251

Please sign in to comment.