Hi,
I have a basic conceptual question that I don’t understand.
Why when working with cuda do I need to move my model to cuda device and also the X and y tensors to cuda device?
at what scenario, in case I’m working with cuda, I would move for example my model to cuda device but keep on working with tensors on CPU?
Is there a simple way to set for example all working torch objects to work with cuda?
something like:
torch.cuda = True

and it will send all relevant object to cuda device automatically ?
Thanks in advance.