Yeah, I’ve built torchvision from source, where these names were modified (source 4).
You are right, if you need to transform the PIL.Images, my suggestion won’t work.
I had in mind to push the already transformed tensors onto the GPU.
However, I’m not even sure if you’ll really see a performance benefit compared to push the data onto the GPU in your training loop using multiple workers.