search

yossigandelsman / DoubleDIP

Official implementation of the paper "Double-DIP: Unsupervised Image Decomposition via Coupled Deep-Image-Priors"
513 stars 95 forks source link

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.30 GiB already allocated; 13.37 MiB free; 29.76 MiB cached) #6

Closed mahendrabvala closed 5 years ago

mahendrabvala commented 5 years ago

Traceback (most recent call last):

File "", line 1, in runfile('E:/Projects/AI/DoubleDIP-master/watermarks_removal.py', wdir='E:/Projects/AI/DoubleDIP-master')

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 827, in runfile execfile(filename, namespace)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 110, in execfile exec(compile(f.read(), filename, 'exec'), namespace)

File "E:/Projects/AI/DoubleDIP-master/watermarks_removal.py", line 567, in remove_watermark_many_images(['f1', 'f2', 'f3'], [im1, im2, im3], "fotolia_many_images")

File "E:/Projects/AI/DoubleDIP-master/watermarks_removal.py", line 544, in remove_watermark_many_images s.optimize()

File "E:/Projects/AI/DoubleDIP-master/watermarks_removal.py", line 440, in optimize self._optimization_closure(j, step)

File "E:/Projects/AI/DoubleDIP-master/watermarks_removal.py", line 503, in _optimization_closure self.mask_net_output = self.mask_net(mask_net_input)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 493, in call result = self.forward(*input, **kwargs)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\container.py", line 92, in forward input = module(input)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 493, in call result = self.forward(*input, **kwargs)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\container.py", line 92, in forward input = module(input)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 493, in call result = self.forward(*input, **kwargs)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\container.py", line 92, in forward input = module(input)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 493, in call result = self.forward(*input, **kwargs)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\container.py", line 92, in forward input = module(input)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 493, in call result = self.forward(*input, **kwargs)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\container.py", line 92, in forward input = module(input)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 493, in call result = self.forward(*input, **kwargs)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\container.py", line 92, in forward input = module(input)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 493, in call result = self.forward(*input, **kwargs)

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\modules\padding.py", line 171, in forward return F.pad(input, self.padding, 'reflect')

File "C:\Users\Kinjal\Anaconda3\lib\site-packages\torch\nn\functional.py", line 2817, in pad ret = torch._C._nn.reflection_pad2d(input, pad)

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.30 GiB already allocated; 13.37 MiB free; 29.76 MiB cached)