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

leaf Variable inplace bug fix #1759

Merged
merged 1 commit into from
Dec 23, 2020
Merged

leaf Variable inplace bug fix #1759

merged 1 commit into from
Dec 23, 2020

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Dec 23, 2020

This PR fixes Docker issue with leaf Variable inplace ops in #1552.

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Improved bias initialization in YOLO model.

πŸ“Š Key Changes

  • Modified the code to use b.data instead of b when adjusting bias values during model initialization.

🎯 Purpose & Impact

  • 🎯 Purpose: To strictly modify the tensor data without affecting its grad_fn, thus enhancing the code's consistency and clarity when initializing biases.
  • 🎈 Impact: This tweak ensures that gradient calculations are not inadvertently affected during initialization, potentially leading to more stable and reliable training of the YOLO models for object detection tasks. Users may experience subtle improvements in model training efficiency and performance.

@glenn-jocher glenn-jocher merged commit 9922c11 into master Dec 23, 2020
@glenn-jocher glenn-jocher deleted the leaf_fix branch December 23, 2020 01:27
KMint1819 pushed a commit to KMint1819/yolov5 that referenced this pull request May 12, 2021
taicaile pushed a commit to taicaile/yolov5 that referenced this pull request Oct 12, 2021
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 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