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

add create_calibrator function #211

Merged
merged 3 commits into from
Aug 22, 2022
Merged

add create_calibrator function #211

merged 3 commits into from
Aug 22, 2022

Conversation

Data-Iab
Copy link
Collaborator

@Data-Iab Data-Iab commented Aug 16, 2022

New feature : create_calibrator function that allows to create calibrator from name and kwargs is added to alonet.torch2trt to optimize imports in Tensorrt scripts.

from alonet.torch2trt import create_calibrator, DataBatchStreamer

cache_file = "calib.bin"
data_streamer = DataBatchStreamer(...)

calibrator = create_calibrator("minmax", data_streamer, cache_file)

This pull request includes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • This change requires a documentation update

@Data-Iab Data-Iab self-assigned this Aug 16, 2022
@thibo73800
Copy link
Contributor

You can follow this format for your pull request. (should be automatic next time)
https://github.com/Visual-Behavior/aloception/blob/master/.github/pull_request_template.md

@thibo73800 thibo73800 merged commit 2fbcce1 into master Aug 22, 2022
@thibo73800 thibo73800 deleted the trt branch August 22, 2022 05:59
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.

2 participants