一、问题描述
记录一下:复现(Windows平台下,机器学习算法)代码过程中遇到的多进程问题,报错截图如下。
根据排查定位到代码出错位置349行。(背景介绍:logprog封装了dataloader类,本质上还是在遍历输出data_loader这个实例过程中<torch.utils.data.dataloader.DataLoader>引发的问题)。
二、解决方式
在网上搜索torch.utils.data.dataloader.DataLoader 关于多进程报错,发现是其参数num_workers导致的,将num_workers改成0就解决了。
补充,源码中num_workers的说明如下。
num_workers (int, optional): how many subprocesses to use for data loading. ``0`` means that the data will be loaded in the main process. (default: ``0``)