问题描述:

环境:win10
情况:当时正在运行yolov5程序,出现这个问题

BrokenPipeError: [Errno 32] Broken pipe
在这里插入图片描述

解决方案:

在参数设计阶段将worker设置成0,在win10上非常有必要将nw(num-worker)设置成0


问题分析

worker参数是指的是:
在使用Pytorch的DataLoad函数时,一个重要的参数

  • 数据需要加载到内存中在传入GPU,numworker越大,传入的越快,当然也很吃CPU
  • 在WIN中无比设置成0,在Linux中可以设置成多线程CPU的核心数。

在这里插入图片描述

Logo

更多推荐