[GitHub] [incubator-mxnet] fierceX commented on issue #15716: Increase amp support for Bi-lstm and Concat operators in gluon

[GitBox]

fierceX commented on issue #15716: Increase amp support for Bi-lstm and Concat 
operators in gluon
URL: 
https://github.com/apache/incubator-mxnet/issues/15716#issuecomment-517521798
 
 
   Hi @ptrendx ,The following code should be able to reproduce this error.
   ```python
   import mxnet as mx
   from mxnet import nd
   from mxnet.gluon import nn,rnn
   from mxnet.contrib import amp
   
   model = nn.HybridSequential()
   model.add(rnn.LSTM(hidden_size=10,num_layers=2,bidirectional=True))
   model.add(nn.Dense(2))
   
   model.initialize()
   model.hybridize()
   model(nd.ones((2,3,4)))
   
   new_model = amp.convert_hybrid_block(model)
   ```

----------------------------------------------------------------
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:
us...@infra.apache.org


With regards,
Apache Git Services