So I could reproduce my problem. The link to code is given below :


train + validation :  https://www.kaggle.com/suchith0312/pytorch-dataloader-testing?scriptVersionId=14425205 34


train + validation + test : https://www.kaggle.com/suchith0312/pytorch-dataloader-testing?scriptVersionId=14425182 20


As you can see from the output that train_id is different for two cases which must not happen. My original code is of same format.