search

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏
Apache License 2.0
1.5k stars 446 forks source link

有关yolov4模型稀疏训练问题 #119

Open Joonhow opened 3 years ago

Joonhow commented 3 years ago

请问利用yolov4训练时，需要做代码调整吗。本人在系数训练的过程中适当调小了稀疏速率，然而稀疏训练过程中mAP掉的非常夸张，从66掉到不到1，现在在13左右。数据是COCO的80类，模型是mish改为leaky的yolov4

chumingqian commented 3 years ago

610272BF1AB42D4ED35DDD0B5433E45B @Joonhow @tanluren : 您好，请问问题解决吗？ 我在VOC上稀疏化训练， 稀疏度 sr=0.001， prune=1; 精度可以恢复， 但召回率就很低。

sunshine-zkf commented 3 years ago

@Joonhow 你好，请问你在该版本上从头训练过YOLOv4吗，训练精度如何？我的训练精度只能达到60mAP，稀疏化训练的时候设置全局衰减s = 0.0001进度能到54。

wuyexingkong commented 2 years ago

python train.py --cfg cfg\yolov4_digital.cfg --data data\digital.data --weights weights\yolov4_digital.weights --epochs 20 --batch-size 4 -sr --s 0.001 --prune 1

训练到18个epochs出现内存不足问题： image 各位大佬，有没有什么心得传授一下？