This commit is contained in:
jiahangxu 2021-10-26 16:50:53 +08:00
Родитель 57bb1a5e9a
Коммит e904d1aaa0
3 изменённых файлов: 9 добавлений и 9 удалений

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

@ -29,15 +29,15 @@ jobs:
~/.nn_meter
/home/runner/work/nn-Meter/data/testmodels
/opt/hostedtoolcache/Python/3.6.10/x64/lib/python3.6/site-packages
key: ${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
key: Data-${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
- name: Install dependencies
if: steps.cache.outputs.cache-hit != 'true'
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

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

@ -29,15 +29,15 @@ jobs:
~/.nn_meter
/home/runner/work/nn-Meter/data/testmodels
/opt/hostedtoolcache/Python/3.6.10/x64/lib/python3.6/site-packages
key: ${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
key: Data-${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
- name: Install dependencies
if: steps.cache.outputs.cache-hit != 'true'
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 nni

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

@ -29,15 +29,15 @@ jobs:
~/.nn_meter
/home/runner/work/nn-Meter/data/testmodels
/opt/hostedtoolcache/Python/3.6.10/x64/lib/python3.6/site-packages
key: ${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
key: Data-${{hashFiles('nn_meter/configs/predictors.yaml')}}-${{hashFiles('tests/integration_test.py')}}
- name: Install dependencies
if: steps.cache.outputs.cache-hit != 'true'
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