进行yolov4稀疏化训练

[mingwz]

RTX 3090，ubuntu18.04服务器
-batch-size 4 -sr --s 0.0005

再进行darknet训练出来的yolov4上进行稀疏化训练时，总是会在某个epoch停住，之后显存占比会慢慢的上升。最开始占比为39%，如果你不主动停止程序，他会一直上升。

这是怎么一回事呢？

[mingwz]

难道是我的数据的原因吗？

[wuyexingkong]

python train.py --cfg cfg\yolov4_digital.cfg --data data\digital.data --weights weights\yolov4_digital.weights --epochs 20 --batch-size 4 -sr --s 0.001 --prune 1

训练到第18个epochs报错内存不够，如下所示：

楼主你这个问题解决了吗？

[sticktotheend]

@wuyexingkong 您好，请问使用yolov4/yolov4-tiny，需要修改哪些部分呢？我刚接触这方面的学习，希望您不吝赐教。