03_pytorch_computer_vision #137
Unanswered
radityawibhawa
asked this question in
Q&A
Replies: 2 comments
-
Beta Was this translation helpful? Give feedback.
0 replies
-
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. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
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)
Beta Was this translation helpful? Give feedback.
All reactions