vqa.pytorch icon indicating copy to clipboard operation
vqa.pytorch copied to clipboard

Batch first in LSTM

Open DenisDsh opened this issue 7 years ago • 0 comments

In the LSTM documentation is specified that we should feed to the RNN inputs of shape (seq_len, batch, input_size) however it seems to me that we are feeding inputs with shape (batch, seq_len, input_size).

Therefore I believe that the parameter batch_first should be set True.

DenisDsh avatar Aug 02 '18 10:08 DenisDsh