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
I'm experiencing an issue similar to the one described in this comment. When exporting to ONNX, I encounter an error indicating that tensors are on different devices (CPU and CUDA). The error log is as follows: ERROR:root:Exporting to ONNX failed. 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_CUDA_addmm) ...
Could you please help me understand what might be causing this issue and how I can resolve it? Thank you!