AerialDetection icon indicating copy to clipboard operation
AerialDetection copied to clipboard

about the learning rate for single GPU

Open Amberrferr opened this issue 3 years ago • 2 comments

我想请问一下,根据get start.txt,config的optimizer中的学习率是根据8GPUS设置的,即Batch size为2的情况下,8GPUs的lr为0.01。为何get start.txt中说0.01对应的是4GPUS? 如果我用的是单核GPU,那么我的学习率应该设置为0.00125还是0.0025呢?

Amberrferr avatar Nov 01 '21 15:11 Amberrferr

The learning rate of SGD depends on the batch size. If your batch size is bs, then the lr = 0.01/8 * bs.

dingjiansw101 avatar Nov 02 '21 03:11 dingjiansw101

The learning rate of SGD depends on the batch size. If your batch size is bs, then the lr = 0.01/8 * bs.

Okay, Thanks for you reply!

Amberrferr avatar Nov 02 '21 13:11 Amberrferr