search

LiQiufu / WaveCNet

the code for WaveCNet
123 stars 16 forks source link

RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm) #9

Open Jubbuuliiing opened 2 years ago

Jubbuuliiing commented 2 years ago

when I use this code in my own object,using the dwtlayer and dwtfunction, I met the problem"RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm)" The main problem is located in the class DWTFunction_2D_tiny(Function):L = torch.matmul(matrix_Low_0, input), please tell me what I can do. Thank you very much! R `X(27@WHPXO$7@I5L1M6O

LiQiufu commented 1 year ago

I guess that you didn't add "data.cuda()" for your input data.