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

YOLOv5 AutoCache Update #10027

Merged
merged 7 commits into from
Nov 4, 2022
Merged

YOLOv5 AutoCache Update #10027

merged 7 commits into from
Nov 4, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Nov 3, 2022

@kalenmike @AyushExel YOLOv5 AutoCache PR

Usage is:

python train.py  # no image caching
python train.py --cache   # defaults to 'auto'
python train.py --cache auto  # applies ram caching if sufficient RAM available (50% safety margin)
python train.py --cache ram  # force ram caching
python train.py --cache disk  # force disk caching

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhancement of image caching and memory checks in the YOLOv5 training pipeline.

πŸ“Š Key Changes

  • Clarified the --cache argument help description for image caching options.
  • Introduced memory requirement checks to determine feasibility of caching images in RAM.
  • Changed image caching messages to show cached memory in gigabytes (GB) for better readability.
  • Removed requirement checks from notebook initialization to improve startup time and performance.

🎯 Purpose & Impact

  • Code Clarity: The simpler help message for the --cache argument makes it more understandable to users.
  • Performance Optimization: By checking if there is enough available RAM before caching images, the system prevents potential out-of-memory errors, leading to a smoother user experience.
  • User Information: Displaying cached memory in GB provides a clear indication of how much system resource the process consumes.
  • The removal of the requirement checks from the notebook initialization streamlines the startup process when using YOLOv5 in notebook environments like Jupyter.

@glenn-jocher glenn-jocher changed the title AutoCache YOLOv5 AutoCache Update Nov 3, 2022
@glenn-jocher glenn-jocher merged commit fde7758 into master Nov 4, 2022
@glenn-jocher glenn-jocher deleted the autocache branch November 4, 2022 14:30
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