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

Variable(adv_img_tea).cuda()


变量后面.cuda()