nlp-tutorial icon indicating copy to clipboard operation
nlp-tutorial copied to clipboard

seq2seq_torch maybe have a small mistake

Open wmathor opened this issue 5 years ago • 0 comments

# output : [max_len+1, batch_size, num_directions(=1) * n_hidden]
    output = output.transpose(0, 1) # [batch_size, max_len+1(=6), num_directions(=1) * n_hidden]

to

# output : [max_len+1, batch_size, n_class]
    output = output.transpose(0, 1) # [batch_size, max_len+1(=6), n_class]

wmathor avatar Jun 30 '20 05:06 wmathor