Hello,
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!
Hello,
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!