ZJU-lishuang / yolov5_prune

yolov5 prune,Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
556 stars 136 forks source link

调整三个参数 稀疏度一直不变 #69

Closed rachel-chennn closed 2 years ago

rachel-chennn commented 2 years ago

按照您在其他问题里的回复,我在常规训练之后进行稀疏训练。 但是不管我怎么调整 s(尝试了0.0001到5000),batchsize(尝试了4到16),epoch(尝试了100到200) 训练过程中的可视化都是下图的情况,稀疏度一直不能接近0,请问是什么可能原因呢?

5acbc0f93df814948d5ebb6ae8bfb89

rachel-chennn commented 2 years ago

我的数据集有900张图片

ZJU-lishuang commented 2 years ago

数据集规模小了

rachel-chennn commented 2 years ago

数据集规模小了

请问数据集大概在多少张合适呢

ZJU-lishuang commented 2 years ago

可以参考示例