Linear probe exposure #7
This file contains hidden or 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
| name: Pipeline tests | |
| on: | |
| push: | |
| branches: [main] | |
| pull_request: | |
| branches: [main] | |
| env: | |
| ASPARAGUS_MODELS: "${{ github.workspace }}/models" | |
| #ASPARAGUS: "${{ github.workspace }}/tests/models" | |
| HYDRA_FULL_ERROR: 1 | |
| jobs: | |
| pipeline-tests: | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 20 | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-python@v5 | |
| with: | |
| python-version: "3.11" | |
| cache: pip | |
| - name: Install CPU-only torch then package | |
| run: | | |
| pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu | |
| pip install -e ".[test]" | |
| - name: Run pretrain test | |
| run: | | |
| pytest tests/test_pretrain.py -v --timeout=300 | |
| - name: Run finetune_seg test | |
| run: | | |
| pytest tests/test_finetune_seg.py -v --timeout=300 | |
| - name: Run train_reg test | |
| run: | | |
| pytest tests/test_train_reg.py -v --timeout=300 | |
| - name: Run test_cls test | |
| run: | | |
| pytest tests/test_test_cls.py -v --timeout=300 | |
| - name: Run linear probe test | |
| run: | | |
| pytest tests/test_linear_probe.py -v --timeout=300 | |