Skip to content

Commit 871c1ba

Browse files
committed
export instructions need to be fixed
1 parent c28d39d commit 871c1ba

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

README.md

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,18 +62,28 @@ model = <YOUR MODEL HERE>
6262
x = <YOUR INPUT HERE>
6363

6464
optimized_model = torch_tensorrt.compile(model, example_inputs)
65-
serialize
65+
serialize # fix me
6666
```
6767

6868
#### Step 2: Deploy
6969
##### Deployment in PyTorch:
7070
```python
71+
import torch
72+
import torch_tensorrt
73+
74+
x = <YOUR INPUT HERE>
7175

76+
# fix me
77+
optimized_model = load_model
78+
optimized_model(x)
7279
```
7380

7481
##### Deployment in C++:
7582
```cpp
83+
#include "torch/script.h"
84+
#include "torch_tensorrt/torch_tensorrt.h"
7685

86+
// to fill
7787
```
7888

7989
## Further resources

0 commit comments

Comments
 (0)