Still doesn’t work 
I try to debug to find the problem, and it stuck on the following code(filename: torch/nn/_functions/rnn.py):
output, hy, cy, reserve, new_weight_buf = torch._cudnn_rnn(
            input, weight_arr, weight_stride0,
            flat_weight,
            hx, cx,
            mode, hidden_size, num_layers,
            batch_first, dropout, train, bool(bidirectional),
            list(batch_sizes.data) if variable_length else (),
            dropout_ts)

these are the parameters used above

paras.png1205×427 16.3 KB

can you figure out the reason