coldlarry
coldlarry
normal_prune说的是normal_prune.py,你要在里面修改代码才能支持自己设计的yolo结构能剪枝。
Try to use yolov3-tiny-hand.cfg and yolov3-tiny.weights?I guess you use v3-cfg or v3-weights?
您好,稀疏化Tiny时,使用yolov3-tiny-hand.cfg 这个配置文件。剪枝时使用稀疏化后生成的.weights文件及yolov3-tiny-hand.cfg。
建议设置在0.001和0.005之间,并提高剪枝代码中的percent变量的值。对于正常剪枝,我在四卡上以默认学习率,稀疏因子0.001/0.002稀疏训练,273轮后剪枝,压缩率大概有80%出头。 ------------------ 原始邮件 ------------------ 发件人: "Luke"
肯定有关的,100轮太少。 ------------------ 原始邮件 ------------------ 发件人: "Luke"
您好,目前的代码只针对标准的yolov3和tiny作剪枝,如果自己想更换backbone,需要自己修改一些代码,但难度不是太大,大概看懂就知道怎么改了。
您好,如果您想把剪枝后的yolo拿来再训练,需要cfg文件和.weights文件。我目前电脑里只存了cfg文件,所以您得自己训练一下了。速度很快的,第一步稀疏化,第二部剪枝即可。(不用训baseline)
我的剪枝代码,是基于标准的yolov3及Tiny写的。如果想用自定义的cfg文件训练也没有问题,但是需要在剪枝代码里做改动,但改动我认为不是太大,需要先把我这个看懂,就知道怎么做了。祝您顺利! ------------------ 原始邮件 ------------------ 发件人: "kokozeng"
你好呀作者,我是把你的代码用在了其他非分类模型的剪枝上,使用时并没有修改你的sketch_martrix函数。 刚才说的那个问题,我感觉更有可能是程序无法从640个卷积核丢掉320个卷积核??? 因为当我把程序改为从640个卷积核,剪到剩下480个卷积核时,程序不会报错。 如果是剩下320个卷积,就会报错。 我还尝试随机生成一个张量,去使用sketch_martrix函数,如果剪的比例太大,也会出现这个问题。 想问一下,大佬是否有修改程序解决这种问题的好办法?