`window` variable is not set to the same device as `img1`

Po-Hsun-Su / pytorch-ssim

pytorch structural similarity (SSIM) loss

Other

1.87k stars 364 forks source link

Closed gregjohnso closed 6 years ago

gregjohnso commented 6 years ago

*** RuntimeError: tensors are on different GPUs error when input cuda input is not on gpu 0

Po-Hsun-Su commented 6 years ago

Hi Greg, I have merged your commit 881d21 to address this issue. Thank you for letting me know the issue.

gregjohnso commented 6 years ago

Great! I made a 3D version too, if you want it :)

Po-Hsun-Su commented 6 years ago