jasonDasuantou / yolov8_distillation

轻量化剪枝+蒸馏

3 stars 0 forks source link

prune #2

Open loganwu0526 opened 1 week ago

loganwu0526 commented 1 week ago

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 commented 1 week ago

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`