This commit is contained in:
jiahangxu 2021-11-15 18:00:23 +08:00
Родитель 7bf50f83bc
Коммит b7412705d1
2 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -47,7 +47,7 @@ nn-Meter is a latency predictor of models with type of Tensorflow, PyTorch, Onnx
| Testing Model Type | Requirements |
| :----------------: | :-----------------------------------------------------------------------------------------------------------------------: |
| Tensorflow | `tensorflow==1.15.0` |
| Tensorflow | `tensorflow==2.6.0` |
| Torch | `torch==1.9.0`, `torchvision==0.10.0`, (alternative)[`onnx==1.9.0`, `onnx-simplifier==0.3.6`] or [`nni>=2.4`][1] |
| Onnx | `onnx==1.9.0` |
| nn-Meter IR graph | --- |

Просмотреть файл

@ -22,7 +22,7 @@ nn-Meter is a latency predictor of models with type of Tensorflow, PyTorch, Onnx
| Testing Model Type | Requirements |
| :-------------------: | :------------------------------------------------: |
| Tensorflow | `tensorflow==1.15.0` |
| Tensorflow | `tensorflow==2.6.0` |
| Torch | `torch==1.9.0`, `torchvision==0.10.0`, (alternative)[`onnx==1.9.0`, `onnx-simplifier==0.3.6`] or [`nni>=2.4`][1] |
| Onnx | `onnx==1.9.0` |
| nn-Meter IR graph | --- |