03_pytorch_computer_vision

[radityawibhawa]

trying the functionizing train and test steps, but got this error

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat1 in method wrapper_addmm)

[radityawibhawa]

[mrdbourke]

Hi @radityawibhawa,

Have you gone back through your code and made sure the model + data are on the same device?

If they aren't, you'll get errors like the above.