Torch-Pruning icon indicating copy to clipboard operation
Torch-Pruning copied to clipboard

Published 20 hours ago • verified publisher icon VainF

How to iteratively prune in yolov5?

Open AFallDay opened this issue 1 year ago • 1 comments

When I set the number of iterative pruning to 5, it only performs one pruning and then ends, how do I need to change the code to perform multiple iterative pruning? thanks

AFallDay avatar Jan 23 '24 08:01 AFallDay

Hi @AFallDay, please repeat the pruner.step() for 5 times.

VainF avatar Jan 23 '24 09:01 VainF