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

Add weight decay to argparser #9006

Merged
merged 1 commit into from
Aug 17, 2022
Merged

Add weight decay to argparser #9006

merged 1 commit into from
Aug 17, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Aug 17, 2022

Signed-off-by: Glenn Jocher glenn.jocher@ultralytics.com

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Improved configurability of the weight decay parameter for training in the YOLOv5 classification module.

πŸ“Š Key Changes

  • The optimizer now uses a customizable decay parameter from the command line options.
  • Added a --decay command line argument to specify weight decay.

🎯 Purpose & Impact

  • Purpose: To allow users greater flexibility when setting up the weight decay during model training, which can help in preventing overfitting and improving generalization.
  • Impact: Users can now fine-tune the weight decay according to their dataset and training regimen, potentially leading to better model performance. πŸš€

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
@glenn-jocher glenn-jocher self-assigned this Aug 17, 2022
@glenn-jocher glenn-jocher merged commit 6728dad into master Aug 17, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch August 17, 2022 20:57
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
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

Successfully merging this pull request may close these issues.

None yet

1 participant