Yolo 学习率没有意义?

时间:2021-07-08 12:21:07

标签: yolo darknet

所以在我的 cfg 文件中,我将学习率设置为

learning_rate=0.001
max_batches = 402000
policy=steps
steps=-1,100,2000,30000
scales=.1,10,.1,.001

据我所知,它从 0.001 开始,很快变为 0.0001, 然后在 100 步时变回 0.001,然后在 2000 步时 = 0.0001,然后在 30000 步时它会变为 0.000001。但我在这里

309360: 0.000222, 0.015274 avg, 0.001000 rate, 0.112701 seconds, 4949760 images

并且学习率仍然是 0.001。我错过了什么吗?这不是学习率缩放的工作原理吗?

这是使用暗网的 Yolov3-tiny。

1 个答案:

答案 0 :(得分:0)

正如 AlexeyAB 在手册中所示,您的步长 = max_batches0.8, max_batches0.9。 我更喜欢你像这样设置你的步骤和比例: 步数 = 321600,361800 比例 = .1,.1 在burn_in 之前,您的学习率会很低(有时为 0.000000,因为您将有 10^-15)。在burn_in 之后,它应该成为cfg 中定义的学习率。在 321600 次迭代后,学习率应该增长 10%,这发生在 361800 次迭代后。