My initial shape of the input tensor is [seq_length * batch_size] which is [414, 128] and when the error occurs, my input shape is same [414, 128].
Moreover, when I use torch==0.4.1, the error shows up after one complete epoch. But when I use torch==1.4.0, it shows up before one complete epoch. I don’t know if it is any matter of concern as different version can handle data differently but felt like important to point out.