mobile-yolov5-pruning-distillation icon indicating copy to clipboard operation
mobile-yolov5-pruning-distillation copied to clipboard

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

Open tommy3266 opened this issue 3 years ago • 2 comments

tommy3266 avatar Jun 12 '21 13:06 tommy3266

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

tommy3266 avatar Jun 12 '21 13:06 tommy3266

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

tiger-98 avatar Apr 09 '22 07:04 tiger-98