Skip to content

Conversation

glenn-jocher
Copy link
Member

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

Resolves ultralytics/hub#82

Signed-off-by: Glenn Jocher [email protected]

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Enhanced stability in model updates during training for the YOLOv5 project.

📊 Key Changes

  • Adjusted the Exponential Moving Average (EMA) update function by detaching the model state_dict tensors during the EMA update process.

🎯 Purpose & Impact

  • Purpose: To ensure that the gradients are not calculated or tracked during the update of EMA, improving computational efficiency and reducing potential errors.
  • Impact: Users should experience more stable training as this change will prevent inadvertent influence on gradient calculations from the EMA update step. This contributes to more reliable convergence and performance of the YOLOv5 models. 🚀

@glenn-jocher glenn-jocher self-assigned this Aug 21, 2022
@glenn-jocher glenn-jocher merged commit 841f312 into master Aug 21, 2022
@glenn-jocher glenn-jocher deleted the fix/zero_map branch August 21, 2022 00:23
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
Resolves ultralytics/hub#82

Signed-off-by: Glenn Jocher <[email protected]>

Signed-off-by: Glenn Jocher <[email protected]>
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.

mAP zero while training

2 participants