Skip to content
This repository was archived by the owner on Sep 12, 2025. It is now read-only.

Commit 8fe6940

Browse files
committed
refactor nn_meter_cli
1 parent e44fdc4 commit 8fe6940

File tree

8 files changed

+215
-385
lines changed

8 files changed

+215
-385
lines changed

README.md

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -92,16 +92,16 @@ After installation, a command named `nn-meter` is enabled. To predict the latenc
9292

9393
```bash
9494
# for Tensorflow (*.pb) file
95-
nn-meter --predictor <hardware> [--predictor-version <version>] --tensorflow <pb-file_or_folder>
95+
nn-meter lat_pred --predictor <hardware> [--predictor-version <version>] --tensorflow <pb-file_or_folder>
9696

9797
# for ONNX (*.onnx) file
98-
nn-meter --predictor <hardware> [--predictor-version <version>] --onnx <onnx-file_or_folder>
98+
nn-meter lat_pred --predictor <hardware> [--predictor-version <version>] --onnx <onnx-file_or_folder>
9999

100100
# for torch model from torchvision model zoo (str)
101-
nn-meter --predictor <hardware> [--predictor-version <version>] --torchvision <model-name> <model-name>...
101+
nn-meter lat_pred --predictor <hardware> [--predictor-version <version>] --torchvision <model-name> <model-name>...
102102

103103
# for nn-Meter IR (*.json) file
104-
nn-meter --predictor <hardware> [--predictor-version <version>] --nn-meter-ir <json-file_or_folder>
104+
nn-meter lat_pred --predictor <hardware> [--predictor-version <version>] --nn-meter-ir <json-file_or_folder>
105105
```
106106

107107
`--predictor-version <version>` arguments is optional. When the predictor version is not specified by users, nn-meter will use the latest version of the predictor.
@@ -116,10 +116,10 @@ Furthermore, users may be interested to convert tensorflow pb-file or onnx file
116116

117117
```bash
118118
# for Tensorflow (*.pb) file
119-
nn-meter getir --tensorflow <pb-file> [--output <output-name>]
119+
nn-meter get_ir --tensorflow <pb-file> [--output <output-name>]
120120

121121
# for ONNX (*.onnx) file
122-
nn-meter getir --onnx <onnx-file> [--output <output-name>]
122+
nn-meter get_ir --onnx <onnx-file> [--output <output-name>]
123123
```
124124

125125
Output name is default to be `/path/to/input/file/<input_file_name>_<model-type>_ir.json` if not specified by users.

demo.py

Lines changed: 0 additions & 198 deletions
This file was deleted.

docs/usage.md

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -36,16 +36,16 @@ After installation, a command named `nn-meter` is enabled. To predict the latenc
3636

3737
```bash
3838
# for Tensorflow (*.pb) file
39-
nn-meter --predictor <hardware> [--predictor-version <version>] --tensorflow <pb-file_or_folder>
39+
nn-meter lat_pred --predictor <hardware> [--predictor-version <version>] --tensorflow <pb-file_or_folder>
4040

4141
# for ONNX (*.onnx) file
42-
nn-meter --predictor <hardware> [--predictor-version <version>] --onnx <onnx-file_or_folder>
42+
nn-meter lat_pred --predictor <hardware> [--predictor-version <version>] --onnx <onnx-file_or_folder>
4343

4444
# for torch model from torchvision model zoo (str)
45-
nn-meter --predictor <hardware> [--predictor-version <version>] --torchvision <model-name> <model-name>...
45+
nn-meter lat_pred --predictor <hardware> [--predictor-version <version>] --torchvision <model-name> <model-name>...
4646

4747
# for nn-Meter IR (*.json) file
48-
nn-meter --predictor <hardware> [--predictor-version <version>] --nn-meter-ir <json-file_or_folder>
48+
nn-meter lat_pred --predictor <hardware> [--predictor-version <version>] --nn-meter-ir <json-file_or_folder>
4949
```
5050

5151
`--predictor-version <version>` arguments is optional. When the predictor version is not specified by users, nn-meter will use the latest version of the predictor.
@@ -60,10 +60,10 @@ Furthermore, users may be interested to convert tensorflow pb-file or onnx file
6060

6161
```bash
6262
# for Tensorflow (*.pb) file
63-
nn-meter getir --tensorflow <pb-file> [--output <output-name>]
63+
nn-meter get_ir --tensorflow <pb-file> [--output <output-name>]
6464

6565
# for ONNX (*.onnx) file
66-
nn-meter getir --onnx <onnx-file> [--output <output-name>]
66+
nn-meter get_ir --onnx <onnx-file> [--output <output-name>]
6767
```
6868

6969
Output name is default to be `/path/to/input/file/<input_file_name>_<model-type>_ir.json` if not specified by users.

0 commit comments

Comments
 (0)