训练轮数与Epoch转换关系:根据训练集数量与总batch_size大小计算epoch数,然后将epoch数换算得到训练总轮数max_iters。milestones(学习率变化界限)也是同理。配置文件中batch_size=2,训练集数量为593,训练6个Epoch,在单卡GPU上训练,max_iters=593x6=3558。同理计算milestones为: [2372, 3261]
milestones中的参数一般是max_iters分别除以1.1和1.5得到
下图中第一个是红线是指初始学习率
第二个红线是指学习率调整倍数gamma,设置为0.1
第三个红线是指学习率变化界限milestones,含义为当max_iters达到2372的时候,就会调整学习率,学习率=初始学习率*gamma,当max_iters达到3261时,学习率再次调整。milestones中的参数可以设置多个比如:milestones[2372, 3261,5000]
参考了这篇文章https://www.jianshu.com/p/26a7dbc15246
飞桨7日目标检测课 https://aistudio.baidu.com/aistudio/education/group/info/1617