fix typo based on test v1.0a1
This commit is contained in:
Родитель
d69a52d540
Коммит
e44fdc4096
|
@ -40,12 +40,12 @@ nn-Meter is a latency predictor of models with type of Tensorflow, PyTorch, Onnx
|
||||||
| Testing Model Type | Requirements |
|
| Testing Model Type | Requirements |
|
||||||
| :-------------------: | :------------------------------------------------: |
|
| :-------------------: | :------------------------------------------------: |
|
||||||
| Tensorflow | `tensorflow==1.15.0` |
|
| Tensorflow | `tensorflow==1.15.0` |
|
||||||
| Torch | `torch==1.7.1`, `torchvision==0.8.2`, (alternative)[`onnx==1.9.0`, `onnx-simplifier==0.3.6`] or [`nni==2.4`]$^{[1]}$ |
|
| Torch | `torch==1.7.1`, `torchvision==0.8.2`, (alternative)[`onnx==1.9.0`, `onnx-simplifier==0.3.6`] or [`nni==2.4`][1] |
|
||||||
| Onnx | `onnx==1.9.0` |
|
| Onnx | `onnx==1.9.0` |
|
||||||
| nn-Meter IR graph | --- |
|
| nn-Meter IR graph | --- |
|
||||||
| NNI IR graph | `nni==2.4` |
|
| NNI IR graph | `nni==2.4` |
|
||||||
|
|
||||||
${[1]}$ Please refer to [nn-Meter Usage](#torch-model-converters) for more information
|
[1] Please refer to [nn-Meter Usage](#torch-model-converters) for more information.
|
||||||
|
|
||||||
|
|
||||||
Please also check the versions of `numpy` and `scikit_learn`. The different versions may change the prediction accuracy of kernel predictors.
|
Please also check the versions of `numpy` and `scikit_learn`. The different versions may change the prediction accuracy of kernel predictors.
|
||||||
|
|
|
@ -19,12 +19,12 @@ nn-Meter is a latency predictor of models with type of Tensorflow, PyTorch, Onnx
|
||||||
| Testing Model Type | Requirements |
|
| Testing Model Type | Requirements |
|
||||||
| :-------------------: | :------------------------------------------------: |
|
| :-------------------: | :------------------------------------------------: |
|
||||||
| Tensorflow | `tensorflow==1.15.0` |
|
| Tensorflow | `tensorflow==1.15.0` |
|
||||||
| Torch | `torch==1.7.1`, `torchvision==0.8.2`, (alternative)[`onnx==1.9.0`, `onnx-simplifier==0.3.6`] or [`nni==2.4`]$^{[1]}$ |
|
| Torch | `torch==1.7.1`, `torchvision==0.8.2`, (alternative)[`onnx==1.9.0`, `onnx-simplifier==0.3.6`] or [`nni==2.4`][1] |
|
||||||
| Onnx | `onnx==1.9.0` |
|
| Onnx | `onnx==1.9.0` |
|
||||||
| nn-Meter IR graph | --- |
|
| nn-Meter IR graph | --- |
|
||||||
| NNI IR graph | `nni==2.4` |
|
| NNI IR graph | `nni==2.4` |
|
||||||
|
|
||||||
${[1]}$ Please refer to [nn-Meter Usage](usage.md#torch-model-converters) for more information.
|
[1] Please refer to [nn-Meter Usage](usage.md#torch-model-converters) for more information.
|
||||||
|
|
||||||
Please also check the versions of `numpy` and `scikit_learn`. The different versions may change the prediction accuracy of kernel predictors.
|
Please also check the versions of `numpy` and `scikit_learn`. The different versions may change the prediction accuracy of kernel predictors.
|
||||||
|
|
||||||
|
|
|
@ -1,5 +1,12 @@
|
||||||
# Copyright (c) Microsoft Corporation.
|
# Copyright (c) Microsoft Corporation.
|
||||||
# Licensed under the MIT license.
|
# Licensed under the MIT license.
|
||||||
|
|
||||||
|
try:
|
||||||
|
import pkg_resources # part of setuptools
|
||||||
|
__version__ = pkg_resources.require("nn-meter")[0].version
|
||||||
|
except ModuleNotFoundError:
|
||||||
|
__version__ = 'UNKNOWN'
|
||||||
|
|
||||||
from .nn_meter import (
|
from .nn_meter import (
|
||||||
nnMeter,
|
nnMeter,
|
||||||
load_latency_predictor,
|
load_latency_predictor,
|
||||||
|
|
4
setup.py
4
setup.py
|
@ -4,7 +4,7 @@ from setuptools import setup, find_packages
|
||||||
|
|
||||||
|
|
||||||
setup(
|
setup(
|
||||||
name='nn_meter',
|
name='nn-meter',
|
||||||
version='1.0',
|
version='1.0',
|
||||||
description='nn-Meter is a novel and efficient system to accurately predict the inference latency of DNN models on diverse edge devices.',
|
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 = open('README.md', encoding='utf-8').read(),
|
||||||
|
@ -13,7 +13,7 @@ setup(
|
||||||
author_email='nn-meter-dev@microsoft.com',
|
author_email='nn-meter-dev@microsoft.com',
|
||||||
url='https://github.com/microsoft/nn-Meter',
|
url='https://github.com/microsoft/nn-Meter',
|
||||||
project_urls={
|
project_urls={
|
||||||
'prediction models and fusion rules': 'https://github.com/microsoft/nn-Meter/releases/tag/v1.0-data'
|
'Data of models': 'https://github.com/microsoft/nn-Meter/releases/tag/v1.0-data',
|
||||||
},
|
},
|
||||||
license = 'MIT',
|
license = 'MIT',
|
||||||
classifiers = [
|
classifiers = [
|
||||||
|
|
Загрузка…
Ссылка в новой задаче