sagiebenaim / distancegan Goto Github PK

Hi, I think you have a small bug in the mnist_to_svhn/solver (line 94)

def distance(self, A, B): return torch.abs(torch.mean(A) - torch.mean(B))

should be

def distance(self, A, B): return torch.mean(torch.abs(A - B))

Is there a keras version of DistanceGAN?

I am trying to run my own data using the train.py file which uses the cyclegan architecture. I am trying to run it on a GPU and all libraries are of the same version as you have mentioned. Despite which, I get the following error:
AssertionError: Torch not compiled with CUDA enabled

The command I am running is:
python train.py --dataroot my_data --name exp_1 --model distance_gan --loadSize 128 --fineSize 128 --batchSize 6 --norm instance --A_to_B --nThreads 12 --gpu_ids 0,2

But when I run it with gpu_ids as -1:
python train.py --dataroot my_data --name exp_1 --model distance_gan --loadSize 128 --fineSize 128 --batchSize 6 --norm instance --A_to_B --nThreads 12 --gpu_ids -1
It starts running.

Seems the code was not made for running on GPU??

(/home/hadoop/anaconda2) hadoop@hadoop:~/GAN/DistanceGAN-master$ python ./discogan_arch/distance_gan_model.py --task_name='celebA' --style_A='Eyeglasses' --constraint='Male' --constraint_type=1
Traceback (most recent call last):
File "./discogan_arch/distance_gan_model.py", line 332, in
model.run()
File "./discogan_arch/distance_gan_model.py", line 173, in run
self.initialize()
File "/home/hadoop/GAN/DistanceGAN-master/discogan_arch/disco_gan_model.py", line 142, in initialize
self.test_A = self.test_A.cuda()
File "/home/hadoop/anaconda2/lib/python2.7/site-packages/torch/autograd/variable.py", line 243, in cuda
return CudaTransfer(device_id, async)(self)
File "/home/hadoop/anaconda2/lib/python2.7/site-packages/torch/autograd/_functions/tensor.py", line 160, in forward
return i.cuda(async=self.async)
File "/home/hadoop/anaconda2/lib/python2.7/site-packages/torch/_utils.py", line 65, in cuda
return new_type(self.size()).copy(self, async)
File "/home/hadoop/anaconda2/lib/python2.7/site-packages/torch/cuda/init.py", line 272, in new
_lazy_init()
File "/home/hadoop/anaconda2/lib/python2.7/site-packages/torch/cuda/init.py", line 84, in _lazy_init
_check_driver()
File "/home/hadoop/anaconda2/lib/python2.7/site-packages/torch/cuda/init.py", line 58, in _check_driver
http://www.nvidia.com/Download/index.aspx""")
AssertionError:
Found no NVIDIA driver on your system. Please check that you
have an NVIDIA GPU and installed a driver from
http://www.nvidia.com/Download/index.aspx

i was tried this code on cpu and got the above issue.. Is it possible to run on cpu??
can you suggest me requirement of hardware components and changes required for code from gpu to cpu......