Wohoo, You are right @rasbt, when I do that CPU usage jumps back to 250s. That makes sense as well. Is there better way to implement this in torchvision.transforms?

I see. So you are not using DataLoader or anything like that but just calling that function on each image separately?

Yes