Issue Regarding Pruned-Finetuned Model Size
Closed this issue · 0 comments
anujdutt9 commented
Hi, Thanks for this repo. I have a question. I pruned the original YOLOv5s model with a prob. of 0.3. The size before pruning is roughly 14 MB and after pruning is 8.6 MB.
Now when I am using the train.py script only for finetuning the pruned model, I see that the size of the best.pt model created by the finetuning part is around 19 MB.
Shouldn't the size of the model remain the same? Why has the model size increased?
Any suggestions?
Thanks.