search

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
556 stars 136 forks source link

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

Open maying16 opened 3 years ago

maying16 commented 3 years ago

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

ZJU-lishuang commented 3 years ago

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

ZJU-lishuang commented 3 years ago

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

maying16 commented 3 years ago

谢谢大佬回复。我采用了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)

ZJU-lishuang commented 3 years ago

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

syswyl commented 3 years ago

谢谢大佬回复。我采用了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 commented 1 year ago

谢谢大佬回复。我采用了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)

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

我也是同样的原因