Open Zzzxi opened 10 months ago
Hi, will you consider coming out with yolov8 pruning as a follow-up?
Hi, thanks for your attention to our work. To my knowledge, yolov8 is a bit different from yolov5 for the prunable structures but may still be within the capability of our method. So, if you are interested in pruning yolov8, you could have a try. For us, honestly, we are a bit busy with the coming work these days, so we may validate the effectiveness of pruning yolov8 at the end of this year. Once the experiment is conducted, we will update the comment. At last, thanks again for your interest in our work. Let me know if you have any other questions.
I will try yolov8's pruning, looking forward to your code updates!
any update on yolov8 pruning :) ?
@HankYe 作者有yolov8剪枝相关的更新吗?目前使用该代码对yolov8剪枝出现报错,而对yolov5剪枝正常运行
Hi, we have tested the effectiveness of PAGCP on YOLOv8, and found that the bug comes from the wrong detection of the gradient-based dependency graph for C2f modules. One alternative way to pruning this module is customizing the pruning handler for this module in dependency.py. We will implement this method in the next version after finishing the current work.
Hi, will you consider coming out with yolov8 pruning as a follow-up?