kaushalshetty:

So the arrangement is keras based i.e batches,timesteps,embedding_size that is why i gave batch_size=True.


I would like to challenge that assumption, since you use the pytorch DataLoader to produce the batches. So you ought to check whether it outputs batches,timesteps,embedding_size or timesteps,batches,embedding_size.