@DM24 So i don’t have the code already… but i outlined above how i got it resolved. There’s no other way than debugging it step by step. The first one i did was making sure the inputs they get are the same, so the data generator had the same seed and would generate same batch to both keras and pytorch. Then i evaluated the embedding matrix and saw that in keras i had a different one than i had in pytorch, so i converted the keras one to pytorch and plugged that in and saw that it worked. Hope it makes sense? if you could be more specific or refer to your thread that could be useful. I recommend making a new thread.