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

AutoBatch report include reserved+allocated #9491

Merged
merged 1 commit into from
Sep 19, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Sep 19, 2022

May resolve #9287 (comment)

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

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Improved the batch size estimation in YOLOv5 for better memory utilization.

📊 Key Changes

  • Clarified the comment to specify that it's for YOLOv5 batch size estimation.
  • Updated the formula calculating the fraction of CUDA memory used to include additional terms.

🎯 Purpose & Impact

  • To allow users to better understand that the autobatch function is tailored for YOLOv5.
  • To enhance the accuracy of estimated batch size, thereby improving memory usage efficiency which could potentially reduce the risk of out-of-memory errors and increase performance.
  • Users could see improved results in terms of training speed and system stability when training YOLOv5 models.

@glenn-jocher glenn-jocher merged commit f038ad7 into master Sep 19, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch September 19, 2022 10:18
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.

AutoBatch: CUDA anomaly detected
1 participant