Skip to content

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 requested a review from thibo73800 August 16, 2022 12:24
@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