I have a similar error. But my input size is correct with the sequence length. [32,1,100] and argument length [32]. Do you have any clue what might the problem be? Thanks  Here 10 is my question on this forum