Skip to content

Commit

Permalink
Merge pull request #923 from jaybdub/v0.5.0
Browse files Browse the repository at this point in the history
increment version to v0.5.0
  • Loading branch information
jaybdub authored May 3, 2024
2 parents ac3df64 + e6b4f2c commit 19a3176
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
3 changes: 3 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,9 @@

## [master](https://github.com/NVIDIA-AI-IOT/torch2trt/tree/master)


## [v0.5.0](https://github.com/NVIDIA-AI-IOT/torch2trt/tree/v0.5.0) - 05/3/2024

- Added tensor shape tracking to support dynamic shapes for flatten, squeeze, unsqueeze, view, reshape, interpolate, and getitem methods
- Added EasyOCR example
- Added the ``DatasetRecorder`` context manager, allowing to easily capture of module inputs in large pipeline for calibration and shape inference
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ def trt_lib_dir():

setup(
name='torch2trt',
version='0.4.0',
version='0.5.0',
description='An easy to use PyTorch to TensorRT converter',
packages=find_packages(exclude=exclude_dir),
ext_package='torch2trt',
Expand Down

0 comments on commit 19a3176

Please sign in to comment.