Disable box_nms in frontend test

apache · Jan 28, 2019 · 1544aab · 1544aab
1 parent 8a27c91
commit 1544aab
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 12 deletions.
diff --git a/nnvm/tests/python/frontend/mxnet/test_forward.py b/nnvm/tests/python/frontend/mxnet/test_forward.py
@@ -185,11 +185,6 @@ def test_forward_argmin():
     mx_sym = mx.sym.argmin(data, axis=0)
     verify_mxnet_frontend_impl(mx_sym, (5, 4), (4,))
 
-def test_forward_box_nms():
-    data = mx.sym.var('data')
-    mx_sym = mx.sym.contrib.box_nms(data, topk=-1, force_suppress=True)
-    verify_mxnet_frontend_impl(mx_sym, (3, 100, 6), (3, 100, 6))
-
 def test_forward_slice_axis():
     data = mx.sym.var('data')
     mx_sym = mx.sym.slice_axis(data, axis=1, begin=-5, end=None)
@@ -221,6 +216,5 @@ def test_forward_l2_normalize():
     test_forward_zeros_like()
     test_forward_argmax()
     test_forward_argmin()
-    test_forward_box_nms()
     test_forward_slice_axis()
     test_forward_l2_normalize()
diff --git a/tests/python/frontend/mxnet/test_forward.py b/tests/python/frontend/mxnet/test_forward.py
@@ -190,11 +190,6 @@ def test_forward_argmin():
     mx_sym = mx.sym.argmin(data, axis=0)
     verify_mxnet_frontend_impl(mx_sym, (5, 4), (4,))
 
-def test_forward_box_nms():
-    data = mx.sym.var('data')
-    mx_sym = mx.sym.contrib.box_nms(data, topk=-1, force_suppress=True)
-    verify_mxnet_frontend_impl(mx_sym, (3, 100, 6), (3, 100, 6))
-
 def test_forward_slice_axis():
     data = mx.sym.var('data')
     mx_sym = mx.sym.slice_axis(data, axis=1, begin=-5, end=None)
@@ -227,6 +222,5 @@ def test_forward_l2_normalize():
     test_forward_zeros_like()
     test_forward_argmax()
     test_forward_argmin()
-    test_forward_box_nms()
     test_forward_slice_axis()
     test_forward_l2_normalize()