该问题的产生是由于windows下多线程的问题,和DataLoader类有关。
解决方案:
修改调用torch.utils.data.DataLoader()函数时的 num_workers 参数。该参数官方API解释如下:
该参数是指在进行数据集加载时,启用的线程数目。可以将代码中的num_works修改为0即可。
例:
出错代码:
trainloader = DataLoader(trainset,batch_size=4,shuffle=True,num_workers=2)
修改为:
trainloader = DataLoader(trainset,batch_size=4,shuffle=True,num_workers=0)