Pytorch Imagenet Github Example - Training input variable not cuda?
Why is the input variable not moved to cuda? Thank you!
nirvan
If your input has to use 2 GPUs, it’s more efficient to send the first half of the input to GPU1 and second half to GPU2, rather than sending the entire input to GPU1, and then sending half of it from GPU1 to GPU2. That’s the reason that the input is not transferred to the GPU at the code location that you pointed out.
smth