mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Update pytest for GitHub runner to use --forked with xdist, hopefully eliminate memory buildup

Browse Source
This commit is contained in:
Ross Wightman 2021-12-01 12:09:23 -08:00
parent 78912b6375
commit 58ffa2bfb7
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/tests.yml vendored
View File

@ -30,7 +30,7 @@ jobs:
- name: Install testing dependencies - name: Install testing dependencies
run: | run: |
python -m pip install --upgrade pip python -m pip install --upgrade pip
pip install pytest pytest-timeout expecttest pip install pytest pytest-timeout pytest-xdist expecttest
- name: Install torch on mac - name: Install torch on mac
if: startsWith(matrix.os, 'macOS') if: startsWith(matrix.os, 'macOS')
run: pip install --no-cache-dir torch==${{ matrix.torch }} torchvision==${{ matrix.torchvision }} run: pip install --no-cache-dir torch==${{ matrix.torch }} torchvision==${{ matrix.torchvision }}
@ -48,4 +48,4 @@ jobs:
env: env:
LD_PRELOAD: /usr/lib/x86_64-linux-gnu/libtcmalloc.so.4 LD_PRELOAD: /usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
run: | run: |
pytest -vv --durations=0 ./tests pytest -vv --forked --durations=0 ./tests

1
tests/test_models.py
View File

@ -350,7 +350,6 @@ if 'GITHUB_ACTIONS' in os.environ:
'vit_large*', 'vit_large*',
'vit_base_patch8*', 'vit_base_patch8*',
'xcit_large*', 'xcit_large*',
'*evob', '*evos', # remove experimental evonorm models, seem to cause issues with dtype manipulation
] ]