problem about test

[wgp666]

Traceback (most recent call last):
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 990, in _try_get_data
data = self._data_queue.get(timeout=timeout)
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/multiprocessing/queues.py", line 107, in get
if not self._poll(timeout):
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/multiprocessing/connection.py", line 257, in poll
return self._poll(timeout)
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/multiprocessing/connection.py", line 424, in _poll
r = wait([self], timeout)
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/multiprocessing/connection.py", line 931, in wait
ready = selector.select(timeout)
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/selectors.py", line 415, in select
fd_event_list = self._selector.poll(timeout)
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/site-packages/torch/utils/data/_utils/signal_handling.py", line 66, in handler
_error_if_any_worker_fails()
RuntimeError: DataLoader worker (pid 112301) is killed by signal: Killed.

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

Traceback (most recent call last):
File "main_test_vrt.py", line 376, in
main()
File "main_test_vrt.py", line 82, in main
for idx, batch in enumerate(test_loader):
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1186, in _next_data
idx, data = self._get_data()
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1152, in _get_data
success, data = self._try_get_data()
File "/home/wgp/anaconda3/envs/VRT/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1003, in _try_get_data
raise RuntimeError('DataLoader worker (pid(s) {}) exited unexpectedly'.format(pids_str)) from e
RuntimeError: DataLoader worker (pid(s) 112301) exited unexpectedly