Cadene / Cadene/vqa.pytorch

Batch first in LSTM

Open
#40 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
733
Forks
176
PR merge metrics
No merged PRs in 30d

Description

In the[ LSTM documentation](
https://pytorch.org/docs/stable/nn.html#torch.nn.LSTM) 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**.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.