Pytorch DataLoader问题

运行深度学习的代码时，在数据包装过程中出现了如下的错误：

问题

Traceback (most recent call last):
  File "d:/code/netComprehension_py/main.py", line 45, in <module>
    for step, (b_x, b_y) in enumerate(loader):
  File "D:\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 804, in __next__
    idx, data = self._get_data()
  File "D:\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 771, in _get_data
    success, data = self._try_get_data()
  File "D:\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 737, in _try_get_data
    raise RuntimeError('DataLoader worker (pid(s) {}) exited unexpectedly'.format(pids_str))
RuntimeError: DataLoader worker (pid(s) 27444) exited unexpectedly

数据转化为loader的代码为

loader = Data.DataLoader(dataset=torch_dataset, batch_size=BATCH_SIZE, shuffle=True, num_workers=2)

问题就出现在这个num_workers.

解决

• 直接把num_wokers去掉，即默认值为0
• 把主要的代码放置在

if __name__ == '__main__':
	......

只有放在这个下面，才能实现多线程。