We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 31ef627 commit e1a0c8dCopy full SHA for e1a0c8d
2 files changed
python/README.md
@@ -8,3 +8,11 @@ To run the unit test, please run the following command.
8
python -m unittest test_plugin
9
python -m unittest test_engine
10
```
11
+
12
+## Run TensorRT Engine
13
14
+To run the TensorRT engine, please run the following command.
15
16
+```bash
17
+$ python scripts/run_engine.py
18
+```
python/main.py
@@ -6,8 +6,8 @@
6
7
def main():
- engine_file_path = "data/identity_neural_network.engine"
- plugin_lib_file_path = "build/src/libidentity_conv.so"
+ engine_file_path = "../data/identity_neural_network.engine"
+ plugin_lib_file_path = "../build/src/libidentity_conv.so"
common_runtime.load_plugin_lib(plugin_lib_file_path)
engine = common_runtime.load_engine(engine_file_path)
0 commit comments