search

Syencil / mobile-yolov5-pruning-distillation

mobilev2-yolov5s剪枝、蒸馏，支持ncnn，tensorRT部署。ultra-light but better performence！
MIT License
821 stars 163 forks source link

关于finetune #40

Open 15050188022 opened 3 years ago

15050188022 commented 3 years ago

新手想请教一个比较基础的问题，剪枝后的模型finetune训练次数设置的足够高（比如跟正常训练一样100个epochs）可以取得很不错的mAP，这样做有问题吗？finetune的训练次数有没有什么限制比如只允许在10，20 次这样

Syencil commented 3 years ago

只是没有必要而已 还容易过拟合

yyyooohao commented 3 years ago

只是没有必要而已 还容易过拟合

我也是训练了很多轮，选择最好的权重就可以嘛