Skip to content

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