Solved by someshfengde in post #2 
              
              
                I assume you have to split your dataset into train test and valid format 
for that instead of doing it with the dataloader you have to do it beforehand 
# scikit learns train test split 
from sklearn.model_selection import train_test_split

# I assume you are having some x , y formatted labels 
X_tr…