Skip to content

Commit e1a0c8d

Browse files
committed
Update
1 parent 31ef627 commit e1a0c8d

2 files changed

Lines changed: 10 additions & 2 deletions

File tree

python/README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,3 +8,11 @@ To run the unit test, please run the following command.
88
python -m unittest test_plugin
99
python -m unittest test_engine
1010
```
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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66

77
def main():
88

9-
engine_file_path = "data/identity_neural_network.engine"
10-
plugin_lib_file_path = "build/src/libidentity_conv.so"
9+
engine_file_path = "../data/identity_neural_network.engine"
10+
plugin_lib_file_path = "../build/src/libidentity_conv.so"
1111

1212
common_runtime.load_plugin_lib(plugin_lib_file_path)
1313
engine = common_runtime.load_engine(engine_file_path)

0 commit comments

Comments
 (0)