Torch-Pruning
Torch-Pruning copied to clipboard
关于yolov8的稀疏训练
我的剪枝流程是 1. 先提前测试验证集精度。 2. 稀疏训练10个epoch。 (3.剪枝。 4. 验证剪枝后的精度。 5. finetune并验证精度。重复这三步) 我有以下几个问题:
- 我这剪枝流程是否正确
- 在finetune的时候,是否要加入pruner.regularize(self.model)
- pruner中的参数如iterative_steps和ch_sparsity这些会不会影响稀疏训练,因为这里的iterative_steps=1,而我总的iterative_steps=16。这些参数会对稀疏训练有影响嘛。
期待大佬回答 :>
@VainF 大佬求回复
您好,
- 没问题
- 不用加入pruner.regularize(self.model)
- 会有影响,需要自己调调参
@VainF 大佬,这是我测试剪枝的结果。https://blog.csdn.net/qq_33596242/article/details/133774348 ,您能帮我看看正常不。正常的话,我这流程也可以给其他人做个参考
@VainF 大佬,这是我测试剪枝的结果。https://blog.csdn.net/qq_33596242/article/details/133774348 ,您能帮我看看正常不。正常的话,我这流程也可以给其他人做个参考
结果是正常的,我是他学弟,他自己没完整测过yolo的剪枝
@Reaidu 用我博客里的版本
结果是正常的,我是他学弟,他自己没完整测过yolo的剪枝
师兄你好,请问组剪枝的原理是什么,原论文没有太理解,呜呜呜
感谢感谢!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!