search

hezhangsprinter / DCPDN

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

RuntimeError: DataLoader worker (pid 19597) is killed by signal: Killed #12

Open slc688888 opened 6 years ago

slc688888 commented 6 years ago

when I run train.py , I got en error. Is my memory too little？My memory is 16G and Gpu memory is 8G Details： Traceback (most recent call last): File "train.py", line 210, in val_target_cpu, val_input_cpu = val_target_cpu.float().cuda(), val_input_cpu.float().cuda() File "/home/littlemonster/anaconda3/envs/pic/lib/python2.7/site-packages/torch/_utils.py", line 69, in _cuda return newtype(self.size()).copy(self, async) File "/home/littlemonster/anaconda3/envs/pic/lib/python2.7/site-packages/torch/utils/data/dataloader.py", line 175, in handler _error_if_any_worker_fails() RuntimeError: DataLoader worker (pid 19597) is killed by signal: Killed.

Look forward to your rely！

zhaoxin111 commented 5 years ago

set workers 0

Devin97 commented 5 years ago

It worked thanks!!