Skip to content

Add PyTorch-only benchmark arg#7564

Merged
glenn-jocher merged 1 commit intomasterfrom
glenn-jocher-patch-1
Apr 25, 2022
Merged

Add PyTorch-only benchmark arg#7564
glenn-jocher merged 1 commit intomasterfrom
glenn-jocher-patch-1

Conversation

@glenn-jocher
Copy link
Copy Markdown
Member

@glenn-jocher glenn-jocher commented Apr 25, 2022

Usage:

python utils/benchmarks.py --weights yolov5s.pt --img 640 --pt-only

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Added option to benchmark PyTorch models exclusively.

📊 Key Changes

  • New argument pt_only added to the benchmarking utility.
  • The benchmarking process can be terminated early after testing PyTorch if pt_only is set.
  • CLI now accepts --pt-only as an option to enable this new feature.

🎯 Purpose & Impact

  • 🎯 Purpose: Simplifies benchmarking for users who only want to test the performance of PyTorch models without completing benchmarks for other export formats.
  • Impact: Users can save time and resources during benchmarking when only PyTorch model performance is of interest, making the process more efficient and tailored to specific needs.

Usage:
```
python utils/benchmarks.py --weights yolov5s.pt --img 640 --pt-only
```
@glenn-jocher glenn-jocher merged commit e54e758 into master Apr 25, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch April 25, 2022 03:03
@glenn-jocher glenn-jocher self-assigned this Apr 25, 2022
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
Usage:
```
python utils/benchmarks.py --weights yolov5s.pt --img 640 --pt-only
```
ctjanuhowski pushed a commit to ctjanuhowski/yolov5 that referenced this pull request Sep 8, 2022
Usage:
```
python utils/benchmarks.py --weights yolov5s.pt --img 640 --pt-only
```
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