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() improved reporting #8887

Merged
merged 1 commit into from
Aug 6, 2022
Merged

Conversation

glenn-jocher
Copy link
Member

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

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Refinement of optimizer logging and argument naming in YOLOv5 training code.

πŸ“Š Key Changes

  • Removed the logging of scaled weight decay from train.py.
  • Renamed weight_decay parameter to decay in the smart_optimizer function to maintain consistency.
  • Updated logging in smart_optimizer to provide clearer information including learning rate and weight decay.

🎯 Purpose & Impact

  • The changes streamline the codebase, making it cleaner by removing unnecessary logging information.
  • The renaming of the parameter to decay clarifies its purpose and creates a more intuitive interface.
  • Enhanced logging when creating the optimizer gives users clear visibility of the learning rate (lr) and decay values for different optimizer parameter groups, improving the transparency of the model training process.
  • Overall, these adjustments lead to a more comprehensible and maintainable codebase with minimal direct impact on end users' daily work, but with improvements to the clarity and quality of information available during training.

@glenn-jocher glenn-jocher self-assigned this Aug 6, 2022
@glenn-jocher glenn-jocher changed the title Update smart_optimizer() weight_decay reporting smart_optimizer() weight_decay reporting Aug 6, 2022
@glenn-jocher glenn-jocher changed the title smart_optimizer() weight_decay reporting smart_optimizer() improved reporting Aug 6, 2022
@glenn-jocher glenn-jocher merged commit 20f1b7e into master Aug 6, 2022
@glenn-jocher glenn-jocher deleted the update/smart_optimizer branch August 6, 2022 09:30
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
Update smart_optimizer() weight_decay reporting
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