DataLoader thread killed

[huk112739]

Hello! I am experiencing some issues while using the DiffusionReg project and would appreciate your assistance and advice.

When running the training script, I encounter problems where the DataLoader workers exit unexpectedly and lead to segmentation faults. Specifically, during the data loading phase, the program fails to proceed normally, and I receive the following error messages:
`Data exists. Loading...
Data exists. Loading...
Data exists. Loading...
0%| | 0/1196 [00:00<?, ?it/s]ERROR: Unexpected segmentation fault encountered in worker.
ERROR: Unexpected segmentation fault encountered in worker.
ERROR: Unexpected segmentation fault encountered in worker.
ERROR: Unexpected segmentation fault encountered in worker.
ERROR: Unexpected segmentation fault encountered in worker.
ERROR: Unexpected segmentation fault encountered in worker.
0%| | 0/1196 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/home/sun/anaconda3/envs/diff/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 1120, in _try_get_data
data = self._data_queue.get(timeout=timeout)
File "/home/sun/anaconda3/envs/diff/lib/python3.9/queue.py", line 180, in get
self.not_empty.wait(remaining)
File "/home/sun/anaconda3/envs/diff/lib/python3.9/threading.py", line 316, in wait
gotit = waiter.acquire(True, timeout)
File "/home/sun/anaconda3/envs/diff/lib/python3.9/site-packages/torch/utils/data/_utils/signal_handling.py", line 66, in handler
_error_if_any_worker_fails()
RuntimeError: DataLoader worker (pid 975787) is killed by signal: Segmentation fault.

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/home/sun/PycharmProjects/DiffusionReg/train.py", line 162, in
main(opts)
File "/home/sun/PycharmProjects/DiffusionReg/train.py", line 74, in main
for i, data in enumerate(tqdm(train_loader, 0)):
File "/home/sun/anaconda3/envs/diff/lib/python3.9/site-packages/tqdm/std.py", line 1182, in iter
for obj in iterable:
File "/home/sun/anaconda3/envs/diff/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 628, in next
data = self._next_data()
File "/home/sun/anaconda3/envs/diff/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 1316, in _next_data
idx, data = self._get_data()
File "/home/sun/anaconda3/envs/diff/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 1272, in _get_data
success, data = self._try_get_data()
File "/home/sun/anaconda3/envs/diff/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 1133, in _try_get_data
raise RuntimeError('DataLoader worker (pid(s) {}) exited unexpectedly'.format(pids_str)) from e
RuntimeError: DataLoader worker (pid(s) 975787) exited unexpectedly`

I set ` num in DataLoader_ Setting workers' to 0 will still cause this issue

[ricpruss]

Turn you number of n_cores=0. This is a common mistake when people with a GPU try multithread from the CPU.