Use 3 workers for torch tests

This commit is contained in:
Sylvain Gugger 2021-04-23 18:47:46 -04:00
parent ca6b80cadb
commit 81a6c7cd39

View File

@ -145,7 +145,7 @@ jobs:
key: v0.4-torch-{{ checksum "setup.py" }}
paths:
- '~/.cache/pip'
- run: python -m pytest -n 4 --dist=loadfile -s --make-reports=tests_torch ./tests/ | tee tests_output.txt
- run: python -m pytest -n 3 --dist=loadfile -s --make-reports=tests_torch ./tests/ | tee tests_output.txt
- store_artifacts:
path: ~/transformers/tests_output.txt
- store_artifacts: