Is there a better way to fine-tune yolov5s models after pruning

[AFallDay]

I've tried putting the pruning code into train.py, but its detection is only half as good as the original, is there a better way? thanks

[VainF]

Hi @AFallDay, I'm not very familiar with Yolo. Using a smaller learning rate can be helpful.

[maolinLiu666]

I've tried putting the pruning code into train.py, but its detection is only half as good as the original, is there a better way? thanks

Hello, I have also encountered this problem. Do you have a good solution? Please reply. Thank you