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权重的问题 #39

Open mengqingyan97 opened 3 years ago

mengqingyan97 commented 3 years ago

QQ图片20210603192426 QQ图片20210603192431 您好，我稀疏训练后得到的BN权重如图所示，和tanluren/yolov3-channel-and-layer-pruning上的完全不同，请问造成这个问题的原因是什么？是我稀疏训练的不够吗？还是我稀疏训练过程的参数设置有问题？我稀疏训练和基础训练都是100epoch，稀疏训练中s=0.001，prune=1

ZJU-lishuang commented 3 years ago

我提供了例子，如果跑示例正常自己数据集不正常，那就是参数问题。要自己去尝试合适的参数

monsterlyg commented 3 years ago

我的perlayer_bn_weights基本没有变化 从epoch的bn分布来看确实如此 跟你差不多 你用的稀疏训练的版本是哪一个？ 交流一下

CompliceU commented 2 years ago

@mengqingyan97 你好，不知道你这个图是怎么得到的呢

GuanTianQing commented 1 year ago

用tensorboard 查看 events文件就可以看到直方统计图了__ ![Uploading image.png…]()