DataLoader Error in FairNodeEmd

[Jarvis-K]

Hi,
I am running the notebook of 'fair_node_embedding'. But an unexpected error raised:

    760         try:
--> 761             data = self._data_queue.get(timeout=timeout)
    762             return (True, data)

/mnt/data1/jjs/anaconda3/envs/HPIC/lib/python3.7/multiprocessing/queues.py in get(self, block, timeout)
    103                     timeout = deadline - time.monotonic()
--> 104                     if not self._poll(timeout):
    105                         raise Empty

/mnt/data1/jjs/anaconda3/envs/HPIC/lib/python3.7/multiprocessing/connection.py in poll(self, timeout)
    256         self._check_readable()
--> 257         return self._poll(timeout)
    258 

/mnt/data1/jjs/anaconda3/envs/HPIC/lib/python3.7/multiprocessing/connection.py in _poll(self, timeout)
    413     def _poll(self, timeout):
--> 414         r = wait([self], timeout)
    415         return bool(r)

/mnt/data1/jjs/anaconda3/envs/HPIC/lib/python3.7/multiprocessing/connection.py in wait(object_list, timeout)
    919             while True:
--> 920                 ready = selector.select(timeout)
    921                 if ready:

/mnt/data1/jjs/anaconda3/envs/HPIC/lib/python3.7/selectors.py in select(self, timeout)
    414         try:
--> 415             fd_event_list = self._selector.poll(timeout)
    416         except InterruptedError:

/mnt/data1/jjs/anaconda3/envs/HPIC/lib/python3.7/site-packages/torch/utils/data/_utils/signal_handling.py in handler(signum, frame)
     65         # Python can still get and update the process status successfully.
---> 66         _error_if_any_worker_fails()
     67         if previous_handler is not None:

RuntimeError: DataLoader worker (pid 54088) is killed by signal: Segmentation fault. 

During handling of the above exception, another exception occurred:

RuntimeError                              Traceback (most recent call last)
/tmp/ipykernel_53070/3824334877.py in <module>
      9         Y_aux,
     10         randomization[epoch],
---> 11         N,
     12     )

/mnt/data1/jjs/Documents/Fair/FairDrop/utils.py in train_rn2v_adaptive(model, loader, optimizer, device, pos_edge_index_tr, y_aux, rand, N)
    110     total_loss = 0
    111 
--> 112     for pos_rw, neg_rw in loader:
    113 
    114         optimizer.zero_grad()

/mnt/data1/jjs/anaconda3/envs/HPIC/lib/python3.7/site-packages/torch/utils/data/dataloader.py in __next__(self)
    343 
    344     def __next__(self):
--> 345         data = self._next_data()
    346         self._num_yielded += 1
    347         if self._dataset_kind == _DatasetKind.Iterable and \

/mnt/data1/jjs/anaconda3/envs/HPIC/lib/python3.7/site-packages/torch/utils/data/dataloader.py in _next_data(self)
    839 
    840             assert not self._shutdown and self._tasks_outstanding > 0
--> 841             idx, data = self._get_data()
    842             self._tasks_outstanding -= 1
    843 

/mnt/data1/jjs/anaconda3/envs/HPIC/lib/python3.7/site-packages/torch/utils/data/dataloader.py in _get_data(self)
    806         else:
    807             while True:
--> 808                 success, data = self._try_get_data()
    809                 if success:
    810                     return data

/mnt/data1/jjs/anaconda3/envs/HPIC/lib/python3.7/site-packages/torch/utils/data/dataloader.py in _try_get_data(self, timeout)
    772             if len(failed_workers) > 0:
    773                 pids_str = ', '.join(str(w.pid) for w in failed_workers)
--> 774                 raise RuntimeError('DataLoader worker (pid(s) {}) exited unexpectedly'.format(pids_str))
    775             if isinstance(e, queue.Empty):
    776                 return (False, None)

RuntimeError: DataLoader worker (pid(s) 54088, 54108) exited unexpectedly```
Can u help me fix the error?