Skip to content

Time profiler#9

Open
Jeffrey-Joan wants to merge 6 commits intomainfrom
time_profiler
Open

Time profiler#9
Jeffrey-Joan wants to merge 6 commits intomainfrom
time_profiler

Conversation

@Jeffrey-Joan
Copy link
Collaborator

  • Model Loading: Loads a YOLO model from a specified path using the Ultralytics library.
  • Random Input Tensor Creation: Creates a random input tensor with user-specified batch size, height, and width.
  • CUDA Support: Optionally moves the model and input tensor to CUDA if available, enhancing profiling performance.
  • Profiling Model Execution: Uses PyTorch's autograd.profiler to profile the model's execution and provides detailed performance metrics.
  • Profiling Results Output: Prints the profiling results to the console, offering insights into the model's execution time, memory usage, and other performance indicators.

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