search

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
559 stars 137 forks source link

稀疏训练没有效果 bn_weights/hist 不能接近1 #61

Closed zimuyuan closed 2 years ago

zimuyuan commented 3 years ago

我在使用你的代码的过程中，在稀疏训练阶段，150 个 epoch 没有效果 bn_weights/hist 不能接近 1 。似乎没有起到稀疏训练的效果。用的是 代码中的 yolov5s_v4 的网络结构和预训练权重，数据集用的 是coco128 图片比较少，想先试试流程能不能跑通下来。 image

ZJU-lishuang commented 3 years ago

coco128目前稀疏未成功过，推测是数据集规模过小

zimuyuan commented 3 years ago

好的，那我试试其他数据集对比看看。