Skip to content

Add weight decay to argparser#9006

Merged
glenn-jocher merged 1 commit intomasterfrom
glenn-jocher-patch-1
Aug 17, 2022
Merged

Add weight decay to argparser#9006
glenn-jocher merged 1 commit intomasterfrom
glenn-jocher-patch-1

Conversation

@glenn-jocher
Copy link
Copy Markdown
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.

1 participant