search

coldlarry / YOLOv3-complete-pruning

提供对YOLOv3及Tiny的多种剪枝版本，以适应不同的需求。
805 stars 214 forks source link

量化后模型参数文件大小并没有变化 #108

Open StrugglingForBetter opened 2 years ago

StrugglingForBetter commented 2 years ago

@coldlarry @PiseyYou

通过修改models.py中的W_bit和A_bit, 设置为8， 但是用下面的指令训练后，得到的模型参数文件大小并没有发生变化

python3 train.py --data data/oxfordhand.data --batch-size 32 --accumulate 1 --weights weights/yolov3.weights --cfg cfg/yolov3-quantize-hand.cfg

按道理量化后模型会变小，请问如何将flot32的模型转换成int8 的模型？

smboj commented 5 months ago

请问你解决了吗，我运行之后还是float32的，怎么解决呢