Skip to content

Commit

Permalink
[CI] Update MxNet to 1.6.0 with MKL (#5240)
Browse files Browse the repository at this point in the history
  • Loading branch information
icemelon authored Apr 6, 2020
1 parent 799ff35 commit 41b8fd1
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion docker/install/ubuntu_install_mxnet.sh
Original file line number Diff line number Diff line change
Expand Up @@ -20,4 +20,4 @@ set -e
set -u
set -o pipefail

pip3 install mxnet==1.5.0
pip3 install mxnet-mkl==1.6.0
2 changes: 1 addition & 1 deletion tests/python/frontend/mxnet/test_forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,7 @@ def test_forward_rrelu():
data = mx.sym.var('data')
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)
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():
data = mx.sym.var('data')
Expand Down

0 comments on commit 41b8fd1

Please sign in to comment.