ScaledYOLOv4 icon indicating copy to clipboard operation
ScaledYOLOv4 copied to clipboard

请问训练自己的数据大概多少个epoch可以收敛

Open Eleanor456 opened this issue 4 years ago • 1 comments
trafficstars

你好,我有一个数据集,大概6000张图片,25个epoch后p和r值,以及50, 95的map值都很低,请问您训练的时候大概多少个epoch可以收敛呢

Eleanor456 avatar Dec 22 '20 06:12 Eleanor456

这个问题没有办法准确回答,按照Paper中是先做了300个Epoch的预训练,然后又基于预训练又做了finetune。数据集不同,训练epoch也有差异。建议tensorboard观察ap的变化,当ap和loss趋于平稳了就可以停掉了。

DataXujing avatar Dec 22 '20 06:12 DataXujing