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

Support exporting TensorRT serialized engine in CLI tools #326

Merged
merged 6 commits into from
Feb 18, 2022

Conversation

zhiqwang
Copy link
Owner

@zhiqwang zhiqwang commented Feb 18, 2022

We provide a Python CLI tools to export the TensorRT serialized engine

python tools/export_model.py --checkpoint_path [path/to/your/best.pt] --include engine

Close #322

@zhiqwang zhiqwang added documentation Improvements or additions to documentation enhancement New feature or request labels Feb 18, 2022
@codecov
Copy link

codecov bot commented Feb 18, 2022

Codecov Report

Merging #326 (2e8d929) into main (31ab550) will not change coverage.
The diff coverage is n/a.

❗ Current head 2e8d929 differs from pull request most recent head 699ca6b. Consider uploading reports for the commit 699ca6b to get more accurate results

Impacted file tree graph

@@           Coverage Diff           @@
##             main     #326   +/-   ##
=======================================
  Coverage   95.38%   95.38%           
=======================================
  Files          11       11           
  Lines         759      759           
=======================================
  Hits          724      724           
  Misses         35       35           
Flag Coverage Δ
unittests 95.38% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.


Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 31ab550...699ca6b. Read the comment docs.

@zhiqwang zhiqwang merged commit 757e904 into main Feb 18, 2022
@zhiqwang zhiqwang deleted the trt-export-cli branch February 18, 2022 06:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
documentation Improvements or additions to documentation enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

TensorRT C++ Example Error
1 participant