sxjscience commented on issue #10453: Bug of CuDNN RNN with variable sequence 
   The following code, which always use `seq_len=500` will not trigger the seg 
fault. This is a very critical bug.
   from mxnet.gluon.rnn import LSTM
   import mxnet as mx
   import numpy as np
   ctx = mx.gpu()
   lstm = LSTM(num_layers=1, hidden_size=200, dropout=0.0)
   batch_size = 32
   for seq_len in range(500, 10, -1):
       for repeat in range(10):
           real_seq_len = 500
           print(real_seq_len, repeat)
           inputs_nd = mx.nd.random.normal(0, 1, shape=(real_seq_len, 
batch_size, 200), ctx=ctx)
           out = lstm(inputs_nd)

This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:

With regards,
Apache Git Services

Reply via email to