Solved by ptrblck in post #2 
              
              
                kf.split will return the train and test indices as far as I know. 
Currently you are passing these indices to a DataLoader, which will just return a batch of indices. 
I think you should pass the train and test indices to a Subset to create new Datasets and pass these to the DataLoaders. 
Let me kno…