pytorch debug 出现 'Broken pipe' 错误

出现错误的原因是因为Dataloader的多线程问题

Google结果显示是Pytorch在win10中的debug,暂不支持多线程,把num_worker改为0问题解决;
查看官方文档结果如下:
image
也是建议我们把num_worker设为0,再逐步Debug检测Code。

另外,针对程序在运行模式下,num_worker>0开启多线程进行加速,程序无法运行的问题,将程序的入口,使用if __name__ == '__main__':包括起来即可。

posted @ 2020-09-18 10:11  wioponsen  阅读(542)  评论(0编辑  收藏  举报