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

yolo.py profiling updates #7178

Merged
merged 2 commits into from
Mar 29, 2022
Merged

yolo.py profiling updates #7178

merged 2 commits into from
Mar 29, 2022

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Mar 28, 2022

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced model profiling options in YOLOv5 for better performance analysis.

πŸ“Š Key Changes

  • πŸ§ͺ Introduced --batch-size argument to specify batch size for profiling.
  • πŸ“ˆ Added --line-profile flag to enable detailed layer-by-layer model profiling.
  • πŸ” Refined the profiling process by utilizing the profile function of torch_utils.
  • πŸš€ Modified profiling to support forward and backward pass performance analysis.
  • 🧹 Removed old Tensorboard code that was not working properly.

🎯 Purpose & Impact

  • πŸ“ Allows users to fine-tune the profiling batch size for their specific hardware resources, optimizing the profiling process.
  • πŸ› οΈ Enables developers to identify performance bottlenecks at the per-layer level, which is critical for targeted model optimization.
  • 🏎️ Gives the option to profile the model's forward and backward passes to assess overall computational performance.
  • 🧽 Codebase cleanup increases maintainability and focuses on functional profiling tools.
  • πŸ€– Enhanced profiling capabilities can lead to better optimized YOLOv5 models, improving speed and efficiency for all users.

@glenn-jocher glenn-jocher self-assigned this Mar 29, 2022
@glenn-jocher glenn-jocher merged commit cf4f3c3 into master Mar 29, 2022
@glenn-jocher glenn-jocher deleted the update/yolo branch March 29, 2022 08:15
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
* yolo.py profiling updates

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
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