BrokenPipeError: [Errno 32] Broken pipe

该问题的产生是由于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)
发布了36 篇原创文章 · 获赞 4 · 访问量 8044

猜你喜欢

转载自blog.csdn.net/qq_41368074/article/details/105560058