Lam1360
Lam1360
@fgilgamesh ,测试的代码应该是没问题的吧,我觉得您可能是pytorch环境没配置好?之前我在跑其他代码的时候也试过这个问题
是的,要根据模型剪枝的结果生成剪枝后的 cfg 文件
直接剔除某一层卷积核的 channel
@violet17 ,CBL_idx 是卷积层+BN+LeakyReLU的简写而已,C是指卷积
@violet17 ,对的,太感谢你了,帮我回答了挺多问题的,我最近都没空回复issue
yolo 层的卷积层?你是指 darknet74 以外的卷积层吗?这些卷积层也有进行剪枝的
哪个卷积层剪枝多少这个是根据训练的效果来确定的
这种现象应该算是正常的,主要看之后能不能 finetune 回去
你预训练权重用的哪个?yolov3 还是 darknet53?
剪枝的时候会先对 bias 做处理,用其后的卷积层或者 BN 层来吸收这个参数,然后再将权重赋值给 compact model,这样做的话就能保证 pruned_model (只对 weight 置 0 的模型)和 Compact_model 对相同的输入有相同的输出