Agreed, In my case, my TX2 has ample GPU to spare, but not CPU. So this serves the purpose as long as the performance isn’t hurt (And looks like it is not in this case) . What I was asking - Is there a way to move the tensor to GPU and then do normalize in torchvision.transforms or is it not meant to do this altogether?
oh and I forgot to thank you, Thank you!!