Skip to content

Commit 81413f2

Browse files
author
Wang
committed
Disable box_nms in frontend test
1 parent b9a09b4 commit 81413f2

File tree

2 files changed

+0
-12
lines changed

2 files changed

+0
-12
lines changed

nnvm/tests/python/frontend/mxnet/test_forward.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -185,11 +185,6 @@ def test_forward_argmin():
185185
mx_sym = mx.sym.argmin(data, axis=0)
186186
verify_mxnet_frontend_impl(mx_sym, (5, 4), (4,))
187187

188-
def test_forward_box_nms():
189-
data = mx.sym.var('data')
190-
mx_sym = mx.sym.contrib.box_nms(data, topk=-1, force_suppress=True)
191-
verify_mxnet_frontend_impl(mx_sym, (3, 100, 6), (3, 100, 6))
192-
193188
def test_forward_slice_axis():
194189
data = mx.sym.var('data')
195190
mx_sym = mx.sym.slice_axis(data, axis=1, begin=-5, end=None)
@@ -221,6 +216,5 @@ def test_forward_l2_normalize():
221216
test_forward_zeros_like()
222217
test_forward_argmax()
223218
test_forward_argmin()
224-
test_forward_box_nms()
225219
test_forward_slice_axis()
226220
test_forward_l2_normalize()

tests/python/frontend/mxnet/test_forward.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -190,11 +190,6 @@ def test_forward_argmin():
190190
mx_sym = mx.sym.argmin(data, axis=0)
191191
verify_mxnet_frontend_impl(mx_sym, (5, 4), (4,))
192192

193-
def test_forward_box_nms():
194-
data = mx.sym.var('data')
195-
mx_sym = mx.sym.contrib.box_nms(data, topk=-1, force_suppress=True)
196-
verify_mxnet_frontend_impl(mx_sym, (3, 100, 6), (3, 100, 6))
197-
198193
def test_forward_slice_axis():
199194
data = mx.sym.var('data')
200195
mx_sym = mx.sym.slice_axis(data, axis=1, begin=-5, end=None)
@@ -227,6 +222,5 @@ def test_forward_l2_normalize():
227222
test_forward_zeros_like()
228223
test_forward_argmax()
229224
test_forward_argmin()
230-
test_forward_box_nms()
231225
test_forward_slice_axis()
232226
test_forward_l2_normalize()

0 commit comments

Comments
 (0)