稀疏化训练的问题

Lam1360 / YOLOv3-model-pruning

在 oxford hand 数据集上对 YOLOv3 做模型剪枝（network slimming）

MIT License

1.67k stars 431 forks source link

Open Liqing6668 opened 5 years ago

Liqing6668 commented 5 years ago

您好，请问我在稀疏化训练的时候，precision逐渐从0增到0.5-0.6左右（第九个epoch），逐渐又开始下降到0.3，现在还在训练，不知道这种现象正常吗？或者稀疏化训练precision正常变化趋势是怎样的。（补充：我的数据集较大，有近两万张图片）

Lam1360 commented 5 years ago

这种现象应该算是正常的，主要看之后能不能 finetune 回去

Liqing6668 commented 5 years ago

还想问您一下，稀疏率的问题，我看您给的是0.01，https://github.com/talebolano/yolov3-network-slimming这个代码给的0.0001，这个应该如何选择是好呢？

IsDora commented 5 years ago

@Liqing6668 请问你的pricision正常了吗，我是参考的链接里的那个工程进行的稀疏化训练，稀疏率设置为0.001，我的样本有13000多，precision的变化跟你的很像，请问你是根据什么指标结束的稀疏化训练呢，最后的precision是多少呢，finetune回去了吗？

Liqing6668 commented 5 years ago

precision不同模型可能不太一样，我大概训练了40个epoch、losses稳定后结束的训练，finetune后精度损失较大。

weixiaolian21 commented 5 years ago

您好，请问我在稀疏化训练的时候，precision逐渐从0增到0.5-0.6左右（第九个epoch），逐渐又开始下降到0.3，现在还在训练，不知道这种现象正常吗？或者稀疏化训练precision正常变化趋势是怎样的。（补充：我的数据集较大，有近两万张图片）

请问最后训练结果怎么样？我稀疏训练时也出现这个问题，recall值很高，但是pricision很低只有0.3左右能加个qq交流交流吗我qq：462181979