0


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%。

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


本文转载自: https://blog.csdn.net/Goyavae/article/details/129889592
版权归原作者 Goyavae 所有, 如有侵权,请联系我们删除。

“YOLOv5训练速度慢 GPU占用率低”的评论:

还没有评论