Syencil/mobile-yolov5-pruning-distillation

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

tommy3266 opened this issue · 2 comments

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

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