vqa.pytorch
vqa.pytorch copied to clipboard
Batch first in LSTM
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.