search

xuebinqin / BASNet

Code for CVPR 2019 paper. BASNet: Boundary-Aware Salient Object Detection
MIT License
1.35k stars 249 forks source link

RuntimeError: DataLoader worker exited unexpectedly #32

Closed aviaisr closed 4 years ago

aviaisr commented 4 years ago

Hi, I'm trying to run the test code but i'm getting the following error: File "C:/CP/DL_REPOSITORIES/BASNet/basnet_test.py", line 76, in for i_test, data_test in enumerate(test_salobj_dataloader): File "C:\ProgramData\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 345, in next data = self._next_data() File "C:\ProgramData\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 841, in _next_data idx, data = self._get_data() File "C:\ProgramData\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 808, in _get_data success, data = self._try_get_data() File "C:\ProgramData\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 774, in _try_get_data raise RuntimeError('DataLoader worker (pid(s) {}) exited unexpectedly'.format(pids_str)) RuntimeError: DataLoader worker (pid(s) 16732) exited unexpectedly

Can you please help?

aviaisr commented 4 years ago

Problem is solved when I changed num_workers to 0.

test_salobj_dataloader = DataLoader(test_salobj_dataset, batch_size=1,shuffle=False,num_workers=1) to: test_salobj_dataloader = DataLoader(test_salobj_dataset, batch_size=1,shuffle=False,num_workers=0)