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

TFDWConv() depthwise_initializer fix #7845

Merged
merged 1 commit into from
May 16, 2022
Merged

TFDWConv() depthwise_initializer fix #7845

merged 1 commit into from
May 16, 2022

Conversation

glenn-jocher
Copy link
Member

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

Resolves bug in #7824

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Improved TensorFlow kernel initialization in YOLOv5 models.

πŸ“Š Key Changes

  • Changed kernel_initializer to depthwise_initializer in the TensorFlow convolution layer configuration.

🎯 Purpose & Impact

  • 🎯 Purpose: Ensures the correct initializer is used when TensorFlow layers are created, specifically for depthwise convolutions which require depthwise_initializer.
  • πŸ’₯ Impact: This change could lead to better performance and stability during the training of YOLOv5 models with TensorFlow, as it allows for the correct initialization of these layers' weights. It might also fix any potential bugs previously present due to the incorrect use of kernel_initializer for depthwise layers.

@glenn-jocher glenn-jocher self-assigned this May 16, 2022
@glenn-jocher glenn-jocher merged commit b52fd48 into master May 16, 2022
@glenn-jocher glenn-jocher deleted the update/tfdwconv branch May 16, 2022 21:37
tdhooghe pushed a commit to tdhooghe/yolov5 that referenced this pull request Jun 10, 2022
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 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.

None yet

1 participant