Will this work on CPU also?

RaphaelMeudec / deblur-gan

Keras implementation of "DeblurGAN: Blind Motion Deblurring Using Conditional Adversarial Networks"

583 stars 244 forks source link

Closed princebhatia09 closed 6 years ago

RaphaelMeudec commented 6 years ago

I ran the deblurring of the images on CPU only, so it should work. However, training requires GPU

princebhatia09 commented 6 years ago

Actually i was thinking to train it on CPU? How can i do that?

RaphaelMeudec commented 6 years ago

It would be impossible to train it on CPU:

fit all images, networks in memory and run computations would be too much and won't fit in memory
even if it fits, training time on GPU was around 10-12 hours, so it would take days

princebhatia09 commented 6 years ago

Can you fix it for CPU version also? I have me and other people like me who are using CPU, that will be a great learning if you can fix it for CPU.

I am using data set of 2000 images only. I think it will fit. Time is not a constraint

RaphaelMeudec commented 6 years ago

There is no difference, you can try to run it on CPU as it is