Is there a version of prunned script wich is compatible with ultralytics 8.5.2?

[BenLag2906]

Dear all,

We are suprised about performance of your work !

We wonder is there is a possiblity to convert our yoloV8.pt train with ultralytics 8.5.2.

We tested on our model and many error are thrown, due to different version of ultralytics

We tested to convert our .pt to .onnx and regenerate a .pt with your checkout tag ( your ultralytic version )

Thanks in advance for your answers !

Best regards

[Christian-lyc]

You have to make some adaption on the ultralytics model to make the model truely pruned, otherwise the pruned model can not be fine-tuned, here's my implementation, https://github.com/Christian-lyc/model_compression-citizen-science-/blob/main/str_prune/stru_prune.py