Solved by NicoHambauer in post #10 
              
              
                I just posted on another thread, but i think this one answers this question as well and may answer it best: 
I would also consider going one level above the ImageFolder Class which inherits from DatasetFolder. 
DatasetFolder uses a method to index the folder subdirectories for each class: 
def find_…