This code does not work if the model have cudnn.BLSTM. Anyone know how to convert a cudnn.BLSTM block to nn.LSTM block