Hello everybody,
I’m following the tutorial about TRAINING ON GPU my neural network, and I want to push my model, and inputs/labels to the GPU but, I have a mistake, and I don’t understand why it crushes!!
We can see in the exercise that we can push to the GPU with theses command lines:
net.to(device)
inputs, labels = inputs.to(device), labels.to(device)
and with my personnal program, it crushes, as with the downloaded file!!
“RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 'weight’”
Can you help me please !