search

DataXujing / ScaledYOLOv4

:fire::fire::fire: Scaled-YOLOv4训练自己的数据集详细教程PDF，关于paper解读请联系小编获取PDF文档
73 stars 24 forks source link

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

Open Eleanor456 opened 3 years ago

Eleanor456 commented 3 years ago

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

DataXujing commented 3 years ago

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