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

Update autobatch.py #5538

Merged
merged 3 commits into from
Nov 6, 2021
Merged

Update autobatch.py #5538

merged 3 commits into from
Nov 6, 2021

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Nov 6, 2021

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced memory reporting in YOLOv5 autobatching with added device name and more precise metrics.

πŸ“Š Key Changes

  • Now includes the device properties, specifically the GPU name, in the memory reporting output.
  • Memory metrics (total, reserved, allocated, and free) are now reported in GiB (Gibibytes) with two decimal places for precision.
  • Formatted the print statement to improve readability and information detail.

🎯 Purpose & Impact

  • This update is intended to give users a clearer understanding of their system's GPU memory usage during the autobatching process.
  • Improved precision in memory reporting can help users make informed decisions when allocating resources for model training and inference.
  • Including device name will immediately clarify which device's memory is being reported, useful for systems with multiple GPUs.
  • Overall, these enhancements can contribute to better resource management and potentially smoother and more efficient model operations.

@glenn-jocher glenn-jocher merged commit cb18cac into master Nov 6, 2021
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch November 6, 2021 12:49
@glenn-jocher glenn-jocher self-assigned this Nov 6, 2021
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
* Update autobatch.py

* Update autobatch.py

* Update autobatch.py
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