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

smart_optimizer() revert to weight with decay #9817

Merged
merged 1 commit into from
Oct 16, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Oct 16, 2022

If a parameter does not fall into any other category

Signed-off-by: Glenn Jocher [email protected]

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Enhanced parameter grouping in optimizer setup for YOLOv5 models.

📊 Key Changes

  • Refactored the way parameters are added to optimizer groups, simplifying the code.
  • Switched to using named_parameters method to iterate over parameters, which is a more general approach.
  • Parameters are now grouped based on their name rather than just their type, allowing for potential custom-named parameters to be correctly categorized for optimization.

🎯 Purpose & Impact

  • Purpose: To ensure that all model parameters are correctly grouped for optimization, with or without weight decay, based on their role in the model (e.g., weights or biases).
  • Impact: This change leads to more ensure comprehensive optimization of all model parameters, potentially improving model training efficiency and accuracy. This could benefit users by providing better-performing models with possibly faster convergence times.

If a parameter does not fall into any other category

Signed-off-by: Glenn Jocher <[email protected]>
@glenn-jocher glenn-jocher merged commit e42c89d into master Oct 16, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-2 branch October 16, 2022 18:51
@glenn-jocher glenn-jocher self-assigned this Oct 16, 2022
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