Skip to content

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.

1 participant