We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c28d39d commit 871c1baCopy full SHA for 871c1ba
README.md
@@ -62,18 +62,28 @@ model = <YOUR MODEL HERE>
62
x = <YOUR INPUT HERE>
63
64
optimized_model = torch_tensorrt.compile(model, example_inputs)
65
-serialize
+serialize # fix me
66
```
67
68
#### Step 2: Deploy
69
##### Deployment in PyTorch:
70
```python
71
+import torch
72
+import torch_tensorrt
73
+
74
+x = <YOUR INPUT HERE>
75
76
+# fix me
77
+optimized_model = load_model
78
+optimized_model(x)
79
80
81
##### Deployment in C++:
82
```cpp
83
+#include "torch/script.h"
84
+#include "torch_tensorrt/torch_tensorrt.h"
85
86
+// to fill
87
88
89
## Further resources
0 commit comments