Skip to content

Commit 7a48317

Browse files
committed
release 0.5.0
1 parent 6659898 commit 7a48317

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

README.md

+5-7
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,11 @@
22

33
This is a branch of [torch2trt](https://github.com/NVIDIA-AI-IOT/torch2trt) with dynamic input support
44

5-
Not all layers support dynamic input such as `torch.split()` etc...
6-
7-
You can create a custom layer from nvinfer1::IPluginV2DynamicExt to implement it.
5+
Note that not all layers support dynamic input such as `torch.split()` etc...
86

97
## Usage
108

11-
Below are some usage examples
9+
Here are some examples
1210

1311
### Convert
1412

@@ -37,7 +35,7 @@ model_trt = torch2trt_dynamic(model, [x], fp16_mode=False, opt_shape_param=opt_s
3735

3836
### Execute
3937

40-
We can execute the returned ``TRTModule`` just like the original PyTorch model
38+
We can execute the returned `TRTModule` just like the original PyTorch model
4139

4240
```python
4341
x = torch.rand(1,3,256,256).cuda()
@@ -79,11 +77,11 @@ python setup.py develop
7977

8078
### Set plugins(optional)
8179

82-
Some layers such as `GN` and `repeat` need c++ plugins. Install the plugin project below
80+
Some layers such as `GN` need c++ plugins. Install the plugin project below
8381

8482
[amirstan_plugin](https://github.com/grimoire/amirstan_plugin)
8583

86-
remember to export the environment variable AMIRSTAN_LIBRARY_PATH
84+
**DO NOT FORGET** to export the environment variable `AMIRSTAN_LIBRARY_PATH`
8785

8886
## How to add (or override) a converter
8987

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def run(self):
130130

131131
setup(
132132
name='torch2trt_dynamic',
133-
version='0.4.1',
133+
version='0.5.0',
134134
description='An easy to use PyTorch to TensorRT converter' +
135135
' with dynamic shape support',
136136
cmdclass={

0 commit comments

Comments
 (0)