YOLOv5训练速度慢 GPU占用率低

   用Yolo5跑自己的数据集,torch_GPU和CUDA都安装了,使用GPU跑模型,GPU的占用率为1%,设置的epoch为300,batch size为32,大概6min多/epoch,数据集很小,刚过一千。在网上查了相关的文章,发现可能是读取数据造成的,把cache-image的action='store_true'改成'store_false'。 

改成

修改后 每epoch大概是13s!超级快,GPU的占用率变到33%,CPU的占用率50%。

后面再想办法改一改,计算机小白学习好难,论文好难(哀嚎~)!

你可能感兴趣的:(深度学习,人工智能,python)