Unverified Commit 41b8fd1e by Haichen Shen Committed by GitHub

[CI] Update MxNet to 1.6.0 with MKL (#5240)

parent 799ff356
...@@ -20,4 +20,4 @@ set -e ...@@ -20,4 +20,4 @@ set -e
set -u set -u
set -o pipefail set -o pipefail
pip3 install mxnet==1.5.0 pip3 install mxnet-mkl==1.6.0
...@@ -125,7 +125,7 @@ def test_forward_rrelu(): ...@@ -125,7 +125,7 @@ def test_forward_rrelu():
data = mx.sym.var('data') data = mx.sym.var('data')
data = mx.sym.concat(data, -data, dim=1) # negative part explicitly data = mx.sym.concat(data, -data, dim=1) # negative part explicitly
mx_sym = mx.sym.LeakyReLU(data, act_type='rrelu', lower_bound=0.3, upper_bound=0.7) mx_sym = mx.sym.LeakyReLU(data, act_type='rrelu', lower_bound=0.3, upper_bound=0.7)
verify_mxnet_frontend_impl(mx_sym, (1, 3, 100, 100), (1, 6, 100, 100)) verify_mxnet_frontend_impl(mx_sym[0], (1, 3, 100, 100), (1, 6, 100, 100))
def test_forward_prelu(): def test_forward_prelu():
data = mx.sym.var('data') data = mx.sym.var('data')
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment