Bump torch from 1.3.1 to 2.2.0 in /.github/requirements #94
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # This workflow will install Python dependencies, run tests and lint with a variety of Python versions | |
| # For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions | |
| name: NEMO | |
| on: | |
| push: | |
| branches: [ master ] | |
| pull_request: | |
| branches: [ master ] | |
| jobs: | |
| build: | |
| runs-on: ubuntu-latest | |
| strategy: | |
| matrix: | |
| python-version: [ "3.5" ] | |
| torch-version: [ "1.3.1", "1.4.0", "1.5.0", "1.6.0", "1.7.1", "1.8.1" ] | |
| steps: | |
| - uses: actions/[email protected] | |
| - name: Set up Python ${{ matrix.python-version }} | |
| uses: actions/[email protected] | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| - name: Install dependencies | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install -r .github/requirements/${{ matrix.torch-version }}.txt | |
| - name: Lint with flake8 | |
| run: | | |
| pip install flake8 | |
| # stop the build if there are Python syntax errors or undefined names | |
| flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics | |
| # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide | |
| flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics | |
| - name: MNIST test | |
| run: | | |
| cd tests; wget https://raw.githubusercontent.com/FrancescoConti/nemo_examples_helper/master/mnist_cnn_fp.pt; PYTHONPATH=`pwd`/.. python mnist_test.py | |
| - name: MobileNet FQ-QD equivalence | |
| run: | | |
| cd tests/mobi_fq_qd_id | |
| wget https://raw.githubusercontent.com/FrancescoConti/nemo_examples_helper/master/mobilenet_1.0_128_best.pth | |
| wget https://raw.githubusercontent.com/FrancescoConti/nemo_examples_helper/master/input_fq.pth | |
| PYTHONPATH=`pwd`/../.. python mobi_fq_qd.py | |
| - name: MobileNet QD-ID equivalence | |
| run: | | |
| cd tests/mobi_fq_qd_id | |
| wget https://raw.githubusercontent.com/FrancescoConti/nemo_examples_helper/master/mobilenet_1.0_128_best.pth | |
| wget https://raw.githubusercontent.com/FrancescoConti/nemo_examples_helper/master/input_fq.pth | |
| wget https://raw.githubusercontent.com/FrancescoConti/nemo_examples_helper/master/mobi_qd_id_res.pth | |
| PYTHONPATH=`pwd`/../.. python mobi_qd_id.py | |