From 1544aab4808d49205c947d71e4c4fa131da37e97 Mon Sep 17 00:00:00 2001 From: Wang Date: Sun, 27 Jan 2019 21:38:50 -0800 Subject: [PATCH] Disable box_nms in frontend test --- nnvm/tests/python/frontend/mxnet/test_forward.py | 6 ------ tests/python/frontend/mxnet/test_forward.py | 6 ------ 2 files changed, 12 deletions(-) diff --git a/nnvm/tests/python/frontend/mxnet/test_forward.py b/nnvm/tests/python/frontend/mxnet/test_forward.py index 70bb4aed39b3a..02d6e0f81fdb9 100644 --- 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 index 45e1cdba0a029..3bdfee89c5dc6 100644 --- 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()