You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
When I modified according to the suggestions, I encountered new problems:
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0
How to solve this problem? thank you.
If you want to train model on CUDA, modify these lines as follows:
Though there are still some lines reporting errors, modify them referring to
Modification2
(i.e.,xx.detach()
=>xx.cpu().detach()
)@TianxiangZhao Please check it if convenient. if correct, a modification sounds good.
The text was updated successfully, but these errors were encountered: