search

Ben-Louis / Deep-Image-Analogy-PyTorch

A python implementation of Deep-Image-Analogy based on pytorch.
MIT License
181 stars 42 forks source link

Expected object of type torch.DoubleTensor but found type torch.cuda.DoubleTensor for argument #2 'other #6

Closed neelkadia-zz closed 6 years ago

neelkadia-zz commented 6 years ago 
sudo python3 main.py
libpng warning: iCCP: known incorrect sRGB profile
patch_size:3; iters:10; rand_d:32
Done All Iterations
patch_size:3; iters:10; rand_d:32
Done All Iterations
12 20 29
Traceback (most recent call last):
  File "main.py", line 49, in <module>
    img_AP, img_B = analogy(img_A, img_BP, config)
  File "/home/kadia/Documents/Deep-Image-Analogy-PyTorch-master/DeepAnalogy.py", line 85, in analogy
    iters=400, display=False)
  File "/home/kadia/Documents/Deep-Image-Analogy-PyTorch-master/VGG19.py", line 142, in get_deconvoluted_feat
    noise, stat = lbfgs(f, noise, maxIter=iters, gEps=1e-4, histSize=4, lr=lr, display=display)
  File "/home/kadia/Documents/Deep-Image-Analogy-PyTorch-master/lbfgs.py", line 60, in lbfgs
    step, stat_ls, args = linesearch(xk.clone(), z, f, fk, gk.clone(), fkm1,gkm1.clone(), 10000, lr)
  File "/home/kadia/Documents/Deep-Image-Analogy-PyTorch-master/lbfgs.py", line 131, in linesearch
    strong_wolfe = torch.DoubleTensor(np.abs(phi_prime_alpha) <= -c2 * phi_prime_0)
RuntimeError: Expected object of type torch.DoubleTensor but found type torch.cuda.DoubleTensor for argument #2 'other'

@Ben-Louis Can you help me in this?