Skip to content

smart_optimizer() revert to weight with decay#9817

Merged
glenn-jocher merged 1 commit intomasterfrom
glenn-jocher-patch-2
Oct 16, 2022
Merged

smart_optimizer() revert to weight with decay#9817
glenn-jocher merged 1 commit intomasterfrom
glenn-jocher-patch-2

Conversation

@glenn-jocher
Copy link
Copy Markdown
Member

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

If a parameter does not fall into any other category

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

🛠️ 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 <glenn.jocher@ultralytics.com>
@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