-
The original YOLOv5 repo calculates the learning rate and schedule automatically based on the given epoch number, batch size, and GPU count. In MMYOLO, default lr is 0.01, but I assume it is calculated for 8 batch size x 8 GPU. How should I scale it in my fine-tuning task when using N GPUs with M batch size per GPU? |
Beta Was this translation helpful? Give feedback.
Answered by
RangeKing
Feb 1, 2023
Replies: 2 comments
-
Do you have any suggestions? @hhaAndroid |
Beta Was this translation helpful? Give feedback.
0 replies
-
Thank you for your attention. Related PR #486. |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
RangeKing
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thank you for your attention. Related PR #486.