Skip to content

leaf Variable inplace bug fix - #1759

Merged
glenn-jocher merged 1 commit into
masterfrom
leaf_fix
Dec 23, 2020
Merged

glenn-jocher merged 1 commit into
masterfrom
leaf_fix

Conversation

@glenn-jocher

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

Copy link
Copy Markdown
Member

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