search

hezhangsprinter / DCPDN

Densely Connected Pyramid Dehazing Network (CVPR'2018)
405 stars 112 forks source link

train.py 在python3.5，pytorch0.3.1运行RuntimeError: #30

Open youyangzhe opened 5 years ago

youyangzhe commented 5 years ago

THCudaCheck FAIL file=/pytorch/torch/lib/THC/generic/THCStorage.cu line=58 error=2 : out of memory Traceback (most recent call last): File "train.py", line 275, in x_hat, tran_hat, atp_hat, dehaze21 = netG(input) File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 325, in call result = self.forward(*input, *kwargs) File "/media/bilibili/D4FA828F299D817A/dujuan/DCPDN-master/models/dehaze22.py", line 728, in forward dehaze=self.relu((self.refine1(dehaze))) File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 325, in call result = self.forward(input, **kwargs) File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/conv.py", line 277, in forward self.padding, self.dilation, self.groups) File "/usr/local/lib/python3.5/dist-packages/torch/nn/functional.py", line 90, in conv2d return f(input, weight, bias) RuntimeError: cuda runtime error (2) : out of memory at /pytorch/torch/lib/THC/generic/THCStorage.cu:58