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

Commit e44fdc4

Browse files
committed
fix typo based on test v1.0a1
1 parent d69a52d commit e44fdc4

File tree

4 files changed

+13
-6
lines changed

4 files changed

+13
-6
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,12 +40,12 @@ nn-Meter is a latency predictor of models with type of Tensorflow, PyTorch, Onnx
4040
| Testing Model Type | Requirements |
4141
| :-------------------: | :------------------------------------------------: |
4242
| Tensorflow | `tensorflow==1.15.0` |
43-
| Torch | `torch==1.7.1`, `torchvision==0.8.2`, (alternative)[`onnx==1.9.0`, `onnx-simplifier==0.3.6`] or [`nni==2.4`]$^{[1]}$ |
43+
| Torch | `torch==1.7.1`, `torchvision==0.8.2`, (alternative)[`onnx==1.9.0`, `onnx-simplifier==0.3.6`] or [`nni==2.4`][1] |
4444
| Onnx | `onnx==1.9.0` |
4545
| nn-Meter IR graph | --- |
4646
| NNI IR graph | `nni==2.4` |
4747

48-
${[1]}$ Please refer to [nn-Meter Usage](#torch-model-converters) for more information
48+
[1] Please refer to [nn-Meter Usage](#torch-model-converters) for more information.
4949

5050

5151
Please also check the versions of `numpy` and `scikit_learn`. The different versions may change the prediction accuracy of kernel predictors.

docs/quick_start.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,12 @@ nn-Meter is a latency predictor of models with type of Tensorflow, PyTorch, Onnx
1919
| Testing Model Type | Requirements |
2020
| :-------------------: | :------------------------------------------------: |
2121
| Tensorflow | `tensorflow==1.15.0` |
22-
| Torch | `torch==1.7.1`, `torchvision==0.8.2`, (alternative)[`onnx==1.9.0`, `onnx-simplifier==0.3.6`] or [`nni==2.4`]$^{[1]}$ |
22+
| Torch | `torch==1.7.1`, `torchvision==0.8.2`, (alternative)[`onnx==1.9.0`, `onnx-simplifier==0.3.6`] or [`nni==2.4`][1] |
2323
| Onnx | `onnx==1.9.0` |
2424
| nn-Meter IR graph | --- |
2525
| NNI IR graph | `nni==2.4` |
2626

27-
${[1]}$ Please refer to [nn-Meter Usage](usage.md#torch-model-converters) for more information.
27+
[1] Please refer to [nn-Meter Usage](usage.md#torch-model-converters) for more information.
2828

2929
Please also check the versions of `numpy` and `scikit_learn`. The different versions may change the prediction accuracy of kernel predictors.
3030

nn_meter/__init__.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,12 @@
11
# Copyright (c) Microsoft Corporation.
22
# Licensed under the MIT license.
3+
4+
try:
5+
import pkg_resources # part of setuptools
6+
__version__ = pkg_resources.require("nn-meter")[0].version
7+
except ModuleNotFoundError:
8+
__version__ = 'UNKNOWN'
9+
310
from .nn_meter import (
411
nnMeter,
512
load_latency_predictor,

setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55

66
setup(
7-
name='nn_meter',
7+
name='nn-meter',
88
version='1.0',
99
description='nn-Meter is a novel and efficient system to accurately predict the inference latency of DNN models on diverse edge devices.',
1010
long_description = open('README.md', encoding='utf-8').read(),
@@ -13,7 +13,7 @@
1313
author_email='nn-meter-dev@microsoft.com',
1414
url='https://github.com/microsoft/nn-Meter',
1515
project_urls={
16-
'prediction models and fusion rules': 'https://github.com/microsoft/nn-Meter/releases/tag/v1.0-data'
16+
'Data of models': 'https://github.com/microsoft/nn-Meter/releases/tag/v1.0-data',
1717
},
1818
license = 'MIT',
1919
classifiers = [

0 commit comments

Comments
 (0)