prune

[loganwu0526]

hi, when I run prune.py, then it just print the model summary, threshold and got stuck, nothing happen.

"YOLOv8-samt_multi3 summary: 321 layers, 9792692 parameters, 0 gradients
tensor(1.0654)"

do you know what happen and how to figure out the bug?

thanks

[jasonDasuantou]

if not error, the model should be saved at res_dir = "./runs/detect/prune/weights/prune.pt"
do you change the dir in the code:
`# Load a model
yolo = YOLO("./runs/detect/yolov8s/weights/last.pt")

Save model address

res_dir = "./runs/detect/prune/weights/prune.pt"

Pruning rate

factor = 0.75`