Hi Soumith,
I would like to know how I can use your solution, if I am using iterator like this
dataiter = iter(trainloader)
images, labels = dataiter.next()

and the trainloader is loading data using torchvision.datasets.ImageFolder
I am stuck on this. I have created my own dataset and I am following CIFAR10 example from pytorch tutorial.
I would like to train a AlexNet using my own dataset.
Thank you in advance.