How to train with cuda in pytorch?

NVlabs / pacnet

Pixel-Adaptive Convolutional Neural Networks (CVPR '19)

https://suhangpro.github.io/pac/

Other

514 stars 79 forks source link

How to train with cuda in pytorch? #6

Open laulampaul opened 5 years ago

laulampaul commented 5 years ago

when I use the pacconvtranspose2d, there is a bug. expected object of backed CUDA but got backend CPU for argument #2 'mat2' How to deal with it? Thank you.

suhangpro commented 5 years ago

Sorry for the late reply. Have you solved it? If not, can you provide more context of the issue (hardware/system info, whether test cases pass, etc.)?