Expected object of backend CPU but got backend CUDA for argument #2 'weight'

Variable(adv_img_tea).cuda()


After the variable .cuda ()

  

Guess you like

Origin www.cnblogs.com/blogwangwang/p/12024495.html