Change export.py workspace argument to MB instead of GB

[DavidBaldsiefen]

Search before asking

• I have searched the YOLOv5 issues and found no similar feature requests.

Description

Currently, export.py has a workspace argument to set the TensorRT workspace size in GB.

yolov5/export.py

Line 220 in 596de6d

config.max_workspace_size = workspace * 1 << 30

However, many GPUs have memory that diverges slightly from the "GB" "rating". For example, a RTX3050 is advertised with 4GB but actually only has 3910MiB of memory. Similar things are true for many other cards. To ensure the user to be able to select a workspace size that reflects this reality, the workspace argument should accept memory sizes in MB

Use case

No response

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!

[github-actions]

👋 Hello, this issue has been automatically marked as stale because it has not had recent activity. Please note it will be closed if no further activity occurs.

Access additional YOLOv5 🚀 resources:

• Wiki – https://github.com/ultralytics/yolov5/wiki
• Tutorials – https://docs.ultralytics.com/yolov5
• Docs – https://docs.ultralytics.com

Access additional Ultralytics ⚡ resources:

• Ultralytics HUB – https://ultralytics.com/hub
• Vision API – https://ultralytics.com/yolov5
• About Us – https://ultralytics.com/about
• Join Our Team – https://ultralytics.com/work
• Contact Us – https://ultralytics.com/contact

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐!