This commit is contained in:
jiahangxu 2021-10-22 18:40:46 +08:00
Родитель 878272c1b8
Коммит 2792d61f4a
3 изменённых файлов: 5 добавлений и 5 удалений

4
.github/workflows/integration-test.yml поставляемый
Просмотреть файл

@ -36,8 +36,8 @@ jobs:
run: |
pip install tensorflow==1.15.0
pip install onnx==1.9.0
pip install torch==1.7.1
pip install torchvision==0.8.2
pip install torch==1.9.0
pip install torchvision==0.10.0
pip install onnx-simplifier
- name: Install nn-Meter

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

@ -36,8 +36,8 @@ jobs:
run: |
pip install tensorflow==1.15.0
pip install onnx==1.9.0
pip install torch==1.7.1
pip install torchvision==0.8.2
pip install torch==1.9.0
pip install torchvision==0.10.0
pip install onnx-simplifier
- name: Install nn-Meter

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

@ -1,6 +1,6 @@
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from nn_meter.utils.utils import try_import_nni, try_import_onnx, try_import_torch, try_import_onnxsim, try_import_nni
from nn_meter.utils.utils import try_import_onnx, try_import_torch, try_import_onnxsim, try_import_nni
import tempfile
from nn_meter.ir_converters.onnx_converter import OnnxConverter