Make workspace size argument use MB #6878

[DavidBaldsiefen]

Closes #6878

To ensure compatibility with existing scripts etc., workspace sizes < 100MB are automatically adjusted to GB

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

🌟 Summary

Improved TensorRT memory workspace configuration for model export in Ultralytics YOLOv5.

📊 Key Changes

• The default TensorRT workspace size has been changed from 4GB to 4096MB.
• Added a check that adjusts the workspace size if a very small value (less than 100MB) is detected.

🎯 Purpose & Impact

• Ensures more flexibility and precision in setting the memory workspace for TensorRT exports.
• Prevents potential issues with configurations that set the workspace size too low, by automatically adjusting it.
• The impact on users includes smoother TensorRT integration and potentially better performance during model export.

[github-actions]

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions YOLOv5 🚀 and Vision AI ⭐.