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

Fix TensorRT exports to ONNX opset 12 #9441

Merged
merged 2 commits into from
Sep 16, 2022
Merged

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Sep 16, 2022

Eliminate opset 13 warning messages:

Screenshot 2022-09-16 at 12 23 55

Signed-off-by: Glenn Jocher [email protected]

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Update to ONNX opset version and logging improvements in TensorRT engine export.

📊 Key Changes

  • Changed the opset version for ONNX export from 13 to 12 during the TensorRT engine export process.
  • Streamlined the logging format for better clarity when displaying input and output names, shapes, and data types.
  • Adjusted the logging message for the FP16/FP32 engine build process to indicate the output file name.

🎯 Purpose & Impact

  • Exporting models using ONNX opset 12 may increase compatibility with different versions of TensorRT, potentially benefiting users working with a specific version that has better support for opset 12.
  • Improved logging contributes to a cleaner and more straightforward output, which enhances the user's ability to understand the export process and debug if necessary.
  • Amended build log provides immediate and clear confirmation of what type of engine was built (FP16 or FP32) and the specific file it was saved as, aiding users in managing their model files more efficiently.

@glenn-jocher glenn-jocher self-assigned this Sep 16, 2022
@glenn-jocher glenn-jocher merged commit 03f2ca8 into master Sep 16, 2022
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch September 16, 2022 10:31
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