sxjscience commented on issue #18918: URL: https://github.com/apache/incubator-mxnet/issues/18918#issuecomment-673699686
@rogerdettloff Actually I cannot reproduce this, would you try to install the nightly version of MXNet and try again? To install the nightly version: ```bash # Install the version with CUDA 10.0 python3 -m pip install -U --pre "mxnet-cu100>=2.0.0b20200802" -f https://dist.mxnet.io/python # Install the version with CUDA 10.1 python3 -m pip install -U --pre "mxnet-cu101>=2.0.0b20200802" -f https://dist.mxnet.io/python # Install the version with CUDA 10.2 python3 -m pip install -U --pre "mxnet-cu102>=2.0.0b20200802" -f https://dist.mxnet.io/python # Install the cpu-only version python3 -m pip install -U --pre "mxnet>=2.0.0b20200802" -f https://dist.mxnet.io/python ``` ```python import mxnet as mx from mxnet import np, npx npx.set_np() net1 = mx.gluon.nn.HybridSequential() net1.add(mx.gluon.nn.Dense(64, activation='relu')) net1.add(mx.gluon.nn.Dense(2)) net1.hybridize() net1.initialize(mx.init.Xavier()) x = mx.np.random.normal(size=(1,64)) out1 = net1(x) net1.export('net1', epoch=1) net2 = mx.gluon.SymbolBlock.imports('net1-symbol.json', ['data'], 'net1-0001.params') out2 = net2(x) print(out2) ``` ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
