PointRCNN作者代码中学习率的变化情况

作者用的是adam onecycle,使用了下面的配置。由于我不想只训练200个epoch rpn阶段,想多训练一些周期,于是自己调整了训练周期,但是这样做训练一段时间之后,学习率还是在上升的,这样肯定是不行的,因为一般训练结束的时候学习率是很小的。

分析学习率变化有两种方法,一种是理论分析,一种看代码分析,一种是试验分析。我采用最后者。但是如果真训练的话又会用太长时间。于是我就把训练的样本从几千个变成了1,这样1S左右一个epoch。

接下来我更改总的学习周期,更改DECAY_STEP_LIST等,看看学习率变化曲线是怎样的。

LR: 0.002
LR_CLIP: 0.00001
LR_DECAY: 0.5
DECAY_STEP_LIST: [100, 150, 180, 200]
LR_WARMUP: True
WARMUP_MIN: 0.0002
WARMUP_EPOCH: 1

下面是DECAY_STEP_LIST: [100, 150, 180, 200],epoches=200

PointRCNN作者代码中学习率的变化情况

 

下面是DECAY_STEP_LIST: [100, 150, 180, 200],epoches=200

PointRCNN作者代码中学习率的变化情况

可以看到好像和设置的 : [100, 150, 180, 200]没有关系,和设置的总训练周期有关系,先上升后下降。

现在改改[100, 150, 180, 200]中的数字个数试试。并没有什么卵用,不太清楚这里的作用。

猜想是训练方法的原因,如果不是adam_onecycle,可能这个列表就有用了。下面试试:

果然。。

DECAY_STEP_LIST: [10, 15, 18, 40] & epoches=100

PointRCNN作者代码中学习率的变化情况

LR_CLIP: 0.00001
LR_DECAY: 0.5
DECAY_STEP_LIST: [10, 15, 18, 20,25,27,28,29,30]
LR_WARMUP: True
WARMUP_MIN: 0.0002
WARMUP_EPOCH: 1

设置文件里面的这些参数应该都是为adam用的。

(一)、什么是Warmup?
Warmup是在ResNet论文中提到的一种学习率预热的方法,它在训练开始的时候先选择使用一个较小的学习率,训练了一些epoches或者steps(比如4个epoches,10000steps),再修改为预先设置的学习来进行训练。引用自学习率预热

 

LR: 0.002 代表最大的学习率。adam和adam_onecycle共用这个参数。