Ultralyics version

[Hemilibeatriz]

Hi Melika!

Which Ultralytics version have you used?

[melika-sce]

Hi @Hemilibeatriz
I used version 8.0.227

[Hemilibeatriz]

I am interested in pruning for the YOLO model, I tried it with its latest version (8.2.2) and it didn't work, and now with version 8.0.227, the error below occurred. Do you know what might be happening? Any idea or help is greatly appreciated!

[melika-sce]

what I see in this picture is 'prune' argument is not implemented in Ultralytics files correctly, so it don't recognize it.
this git would be helpful to start pruning yolov8 models. I don't remember the exact modifications I add to that link. but it helpmed me understand which section I have to modify.
hope this helps!

[Hemilibeatriz]

I see.. I was trying to use the example from your train.py and then do your prune.py example, but since there was an error with the argument, I won't be able to proceed. I thought it was just the Ultralytics version that was the issue.. but there must be more changes then.. anyway, thank you very much for your clarifications!

[melika-sce]

you're welcome 🙌