Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

about the learning rate for single GPU #91

Open
Amberrferr opened this issue Nov 1, 2021 · 2 comments
Open

about the learning rate for single GPU #91

Amberrferr opened this issue Nov 1, 2021 · 2 comments

Comments

@Amberrferr
Copy link

我想请问一下,根据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 Amberrferr changed the title about the learning rate for GPU about the learning rate for single GPU Nov 1, 2021
@dingjiansw101
Copy link
Owner

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

@Amberrferr
Copy link
Author

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!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants