Yeah .
class Model(nn.Module):
    def __init__(self,vocab_len,input_size):
        super(Model,self).__init__()
        self.embed=nn.Embedding(num_embeddings=vocab_len,embedding_dim=128,)
        self.lstm=nn.LSTM(input_size=input_size,hidden_size=1024,num_layers=1)
        self.fc=nn.Linear(in_features=1024,out_features=1)
    def forward(self,x):
        x=self.embed(x)
        o,(ht,ct)=self.lstm(x)
        out=self.fc(ct)
        return nn.Sigmoid(out)

This is the model.
The input shape is (2,560874)[bs,seq_len).
Am I making a blinder?