yolov5_prune icon indicating copy to clipboard operation
yolov5_prune copied to clipboard

Published 20 hours ago • verified publisher icon ZJU-lishuang

您好，请教一下微调时map接近于0的问题

Open maying16 opened this issue 3 years ago • 6 comments

您好，我按照您给出的教程完成了剪枝操作，剪枝前后map分别为0.74，0.72；接下来按照您给出的步骤，用ZJU-lishuang/yolov5-v4中的prune_finetune.py代码进行微调，但是微调训练过程中map值基本保持为0，您知道原因可能有哪些吗？

maying16 avatar Mar 21 '21 07:03 maying16

估计明天会更新蒸馏微调，可以试试效果

ZJU-lishuang avatar Mar 22 '21 09:03 ZJU-lishuang

数据集过少？ 在开源数据集和自己的数据集上试过，结果正常。

ZJU-lishuang avatar Mar 22 '21 09:03 ZJU-lishuang

谢谢大佬回复。我采用了9900多张图像作为训练集，微调时设置为50个epoch，突然发现在第4个epoch中训练损失变成nan了，请问您知道可能是什么导致的吗？（微调训练时有问题，STEP1中的训练和STEP2稀疏训练时都没出现这个问题） Screenshot from 2021-03-23 18-05-21 (2) Screenshot from 2021-03-23 18-05-38 (2)

maying16 avatar Mar 22 '21 10:03 maying16

没有出现过类似的问题，不能定位具体的原因

ZJU-lishuang avatar Mar 23 '21 00:03 ZJU-lishuang

谢谢大佬回复。我采用了9900多张图像作为训练集，微调时设置为50个epoch，突然发现在第4个epoch中训练损失变成nan了，请问您知道可能是什么导致的吗？（微调训练时有问题，STEP1中的训练和STEP2稀疏训练时都没出现这个问题） Screenshot from 2021-03-23 18-05-21 (2) Screenshot from 2021-03-23 18-05-38 (2)

您好，我在实验时候也出现类似问题，请问您发现原因了吗？

syswyl avatar Apr 23 '21 09:04 syswyl

谢谢大佬回复。我采用了9900多张图像作为训练集，微调时设置为50个epoch，突然发现在第4个epoch中训练损失变成nan了，请问您知道可能是什么导致的吗？（微调训练时有问题，STEP1中的训练和STEP2稀疏训练时都没出现这个问题） Screenshot from 2021-03-23 18-05-21 (2) Screenshot from 2021-03-23 18-05-38 (2)

您好，我在实验时候也出现类似问题，请问您发现原因了吗？

我也是同样的原因

Wanghe1997 avatar Oct 22 '22 03:10 Wanghe1997