I will need to send images in batches during training, currently the code works for one image at an instance, my another concern is the code needs image path but when i use imagefolder I dont have any access to image path that i can provide to these transforms, one other way would be to write a custom dataset function where i store image path in a text file along with label and an index that can accesse the pathname of each image and pass it to transforms and make the code work,this is my backup approach. So at the moment i am looking to make it work with ImageFolder alone if possible. Sorry for the long paragraph.