search

Syencil / mobile-yolov5-pruning-distillation

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

使用python3 train.py --type mvocs训练出来的pt模型是20.7M，这正常吗？我看baseline的参数量是3.60 Params(M)，这是为什么呢 #51

Open tommy3266 opened 3 years ago

tommy3266 commented 3 years ago

https://github.com/ultralytics/yolov5/releases，在这里面YOLOv5s的params(M)是7.3，实际训练出来是14.7M，差不多是两倍的关系，

tiger-98 commented 2 years ago

请问你解决了吗？我也是明明mvocs训练出来的模型的参数量只有3.5M，但是最后保存的模型却有28M，推理时间和yolov5s也差不多，没有加速，这是为什么呢