Skip to content

Commit

Permalink
complete v2.0 test
Browse files Browse the repository at this point in the history
  • Loading branch information
JiahangXu committed Jun 27, 2022
1 parent e66d7ef commit cd9bcb3
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion docs/builder/build_kernel_latency_predictor.md
Original file line number Diff line number Diff line change
Expand Up @@ -244,7 +244,7 @@ predictor, data = build_predictor_for_kernel(
)
```

In the experiment of nn-Meter, we set `init_sample_num` as 1000, `finegrained_sample_num` as 10, `iteration` as 5, and `error_threshold` as 0.1.
In the experiment of nn-Meter, we set default `init_sample_num` as 1000, `finegrained_sample_num` as 10, `iteration` as 5, and `error_threshold` as 0.1.

nn-Meter also provided a end-to-end method for users to build a series of general latency predictors, named `nn_meter.builder.build_latency_predictor`. This method will build predictors for all kernels in `<workspace-path>/configs/predictorbuild_config.yaml` according to their corresponding parameters. The parameters includes `INIT_SAMPLE_NUM`, `FINEGRAINED_SAMPLE_NUM`, `ITERATION`, and `ERROR_THRESHOLD`. Here is an example:

Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

setup(
name='nn-meter',
version='2.0a1',
version='2.0',
description='nn-Meter is a novel and efficient system to accurately predict the inference latency of DNN models on diverse edge devices.',
long_description = open('README.md', encoding='utf-8').read(),
long_description_content_type = 'text/markdown',
Expand Down

0 comments on commit cd9bcb3

Please sign in to comment.