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 57 to create new Datasets and pass these to the DataLoaders.
Let me know, if that works for you.