Yes. Just gave it a go and I noticed that torchvision image loading functions load the images to the CPU, and then one has to be copy them to the device (GPU). It would be nice if we can directly load to GPU. Not sure if this would be feasible.