YOLOv5训练速度慢 GPU占用率低
admin 阅读： 2024-03-24
| 后台-插件-广告管理-内容页头部广告（手机） |
用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%。
后面再想办法改一改，计算机小白学习好难，论文好难（哀嚎~）！
声明
1.本站遵循行业规范，任何转载的稿件都会明确标注作者和来源；2.本站的原创文章，请转载时务必注明文章作者和来源，不尊重原创的行为我们将追究责任；3.作者投稿可能会经我们编辑修改或补充。
在线投稿：投稿 站长QQ：1888636
| 后台-插件-广告管理-内容页尾部广告（手机） |
