Hi @skyunyoo, @ptrblck
if we want to use StratifiedKFold for ImageFolder, then we have to pass the labels to split as well.
so it will be something like this :  kf.split(dataset,y)
but for y, how can we assign labels from imagefolder?