Franklin-Zhang0 / Yolo-v8-Apex-Aim-assist

95 stars 20 forks source link

How to prune YOLOv8 model? #26

Open Michelvl92 opened 1 month ago

Michelvl92 commented 1 month ago

Hi,

Do you have an example of how to prune the YOLOv8 models? I already used NVIDIA's https://github.com/NVIDIA/apex/tree/master/apex/contrib/sparsity but this resulted in many errors. I am not necessarily interested in the weights, because I am using this for a different use case, but I am interested in how to perform the YOLOv8 pruning with Nvidia's tool.