环境:windows系统
运行pytorch模型时出现以下报错
BrokenPipeError: [Errno 32] Broken pipe
查阅资料发现该问题的产生是由于windows下多线程的问题,和DataLoader类有关。
解决方法
修改调用torch.utils.data.DataLoader()函数时的 num_workers 参数,该参数是指在进行数据集加载时,启用的线程数目。修改num_works为 0 ,只启用一个主进程加载数据集,避免在windows使用多线程即可。
参考博客:
https://blog.csdn.net/qq_33666011/article/details/81873217