InnovArul:

Maybe, I am a bit late to the party. Why do you think train_id s should not be different?


I am iterating over train only once in an epoch in both the codes. So I expect that the id’s in both cases must be same or I am missing something