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

Variable(adv_img_tea).cuda()


变量后面.cuda()

  

猜你喜欢

转载自www.cnblogs.com/blogwangwang/p/12024495.html